Refactor tests (ashleve#498)

razvanursu · Jan 11, 2023 · 6fcf619 · 6fcf619
1 parent f8b4e12
commit 6fcf619
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,3 +1,5 @@
+"""This file prepares config fixtures for other tests."""
+
 import pyrootutils
 import pytest
 from hydra import compose, initialize
@@ -12,7 +14,7 @@ def cfg_train_global() -> DictConfig:
 
         # set defaults for all tests
         with open_dict(cfg):
-            cfg.paths.root_dir = str(pyrootutils.find_root())
+            cfg.paths.root_dir = str(pyrootutils.find_root(indicator=".project-root"))
             cfg.trainer.max_epochs = 1
             cfg.trainer.limit_train_batches = 0.01
             cfg.trainer.limit_val_batches = 0.1
@@ -35,7 +37,7 @@ def cfg_eval_global() -> DictConfig:
 
         # set defaults for all tests
         with open_dict(cfg):
-            cfg.paths.root_dir = str(pyrootutils.find_root())
+            cfg.paths.root_dir = str(pyrootutils.find_root(indicator=".project-root"))
             cfg.trainer.max_epochs = 1
             cfg.trainer.limit_test_batches = 0.1
             cfg.trainer.accelerator = "cpu"

diff --git a/tests/test_mnist_datamodule.py → tests/test_datamodules.py b/tests/test_mnist_datamodule.py → tests/test_datamodules.py