Merge pull request #62 from ctr26/dev

bioimage_embed/tests/conftest.py

@@ -21,25 +21,11 @@ def dataset(input_dim):
     return FakeData(size=64, image_size=input_dim, num_classes=2, transform=transform)
 
 
-# @pytest.fixture
-# def dataloader_cfg(dataset):
-#     return config.DataLoader(dataset=dataset, num_workers=0)
-
-
 @pytest.fixture
 def lite_model():
     return "dummy_model"
 
 
-# @pytest.fixture
-# def cfg_model(model,input_dim):
-# return config.Model(model=model,input_dim=input_dim)
-
-# @pytest.fixture
-# def cfg(dataloader_cfg,model_cfg):
-#     return config.Config(dataloader=dataloader_cfg, model=model_cfg)
-
-
 @pytest.fixture
 def bie(cfg):
     return BioImageEmbed(cfg)

bioimage_embed/tests/test_cli.py

@@ -43,16 +43,6 @@ def test_write_default_config_file(
     assert config_path.is_file(), "Default config file was not created"
 
 
-# @pytest.fixture
-# def cfg():
-#     mock_dataset = config.ImageFolderDataset(
-#         _target_="bioimage_embed.datasets.FakeImageFolder",
-#     )
-#     cfg = cli.get_default_config()
-#     cfg.recipe.data = mock_dataset
-#     return cfg
-
-
 def test_get_default_config(cfg):
     assert cfg is not None, "Default config should not be None"
     # Further assertions can be added to check specific config properties

bioimage_embed/tests/test_config.py

@@ -36,3 +36,7 @@ def test_model_check(bie):
 
 def test_train_check(bie):
     bie.trainer_check()
+
+
+def test_bie_train(bie):
+    bie.train()

bioimage_embed/tests/test_lightning.py

@@ -12,6 +12,7 @@
 from hydra.utils import instantiate
 from torchvision.datasets import FakeData
 
+
 torch.manual_seed(42)