diff --git a/tests/test_checkpoint.py b/tests/data/test_checkpoint.py
similarity index 100%
rename from tests/test_checkpoint.py
rename to tests/data/test_checkpoint.py
diff --git a/tests/test_dataset.py b/tests/data/test_dataset.py
similarity index 100%
rename from tests/test_dataset.py
rename to tests/data/test_dataset.py
diff --git a/tests/test_sampler.py b/tests/data/test_sampler.py
similarity index 100%
rename from tests/test_sampler.py
rename to tests/data/test_sampler.py
diff --git a/tests/test_module.py b/tests/module/test_module.py
similarity index 100%
rename from tests/test_module.py
rename to tests/module/test_module.py
diff --git a/tests/test_tie_weight.py b/tests/module/test_tie_weight.py
similarity index 100%
rename from tests/test_tie_weight.py
rename to tests/module/test_tie_weight.py
diff --git a/tests/test_callbacks.py b/tests/trainer/test_callbacks.py
similarity index 100%
rename from tests/test_callbacks.py
rename to tests/trainer/test_callbacks.py
diff --git a/tests/test_early_stopping.py b/tests/trainer/test_early_stopping.py
similarity index 100%
rename from tests/test_early_stopping.py
rename to tests/trainer/test_early_stopping.py
diff --git a/tests/test_train_strategy.py b/tests/trainer/test_train_strategy.py
similarity index 100%
rename from tests/test_train_strategy.py
rename to tests/trainer/test_train_strategy.py
diff --git a/tests/test_train_config.py b/tests/trainer/test_trainer.py
similarity index 100%
rename from tests/test_train_config.py
rename to tests/trainer/test_trainer.py