✅ Fix tests

ENSTA-U2IS-AI · Aug 25, 2023 · 588a1a0 · 588a1a0
1 parent a0f1348
commit 588a1a0
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 9 deletions.
diff --git a/tests/models/test_deep_ensembles.py b/tests/models/test_deep_ensembles.py
@@ -25,10 +25,10 @@ def test_list_and_num_estimators(self):
         with pytest.raises(ValueError):
             deep_ensembles([model_1, model_2], num_estimators=2)
 
-    # def test_list_singleton(self):
-    #     model_1 = dummy_model(1, 10, 1)
-    #     with pytest.raises(ValueError):
-    #         deep_ensembles([model_1], num_estimators=1)
+    def test_list_singleton(self):
+        model_1 = dummy_model(1, 10, 1)
+        with pytest.raises(ValueError):
+            deep_ensembles([model_1], num_estimators=1)
 
     def test_model_and_no_num_estimator(self):
         model_1 = dummy_model(1, 10, 1)

diff --git a/tests/routines/test_classification.py b/tests/routines/test_classification.py
@@ -24,7 +24,7 @@ class TestClassificationSingle:
 
     def test_cli_main_dummy_binary(self):
         root = Path(__file__).parent.absolute().parents[0]
-        with ArgvContext("file.py --logits"):
+        with ArgvContext("file.py", "--logits"):
             args = init_args(
                 DummyClassificationBaseline, DummyClassificationDataModule
             )
@@ -46,7 +46,7 @@ def test_cli_main_dummy_binary(self):
 
     def test_cli_main_dummy_ood(self):
         root = Path(__file__).parent.absolute().parents[0]
-        with ArgvContext("file.py --evaluate_ood --entropy"):
+        with ArgvContext("file.py", "--evaluate_ood", "--entropy"):
             args = init_args(
                 DummyClassificationBaseline, DummyClassificationDataModule
             )
@@ -78,7 +78,7 @@ class TestClassificationEnsemble:
 
     def test_cli_main_dummy_binary(self):
         root = Path(__file__).parent.absolute().parents[0]
-        with ArgvContext("file.py --mutual_information"):
+        with ArgvContext("file.py", "--mutual_information"):
             args = init_args(
                 DummyClassificationBaseline, DummyClassificationDataModule
             )
@@ -100,7 +100,7 @@ def test_cli_main_dummy_binary(self):
 
     def test_cli_main_dummy_ood(self):
         root = Path(__file__).parent.absolute().parents[0]
-        with ArgvContext("file.py --evaluate_ood --variation_ratio"):
+        with ArgvContext("file.py", "--evaluate_ood", "--variation_ratio"):
             args = init_args(
                 DummyClassificationBaseline, DummyClassificationDataModule
             )

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -1,4 +1,5 @@
 # fmt: off
+import sys
 from pathlib import Path
 
 import pytest
@@ -47,8 +48,13 @@ def test_cli_main_resnet(self):
     def test_cli_main_other_arguments(self):
         root = Path(__file__).parent.absolute().parents[0]
         with ArgvContext(
-            "file.py --seed 42 --max_epochs 1 --channels_last",
+            "file.py",
+            "--seed",
+            "42",
+            "--max_epochs",
+            "1",
         ):
+            print(sys.orig_argv, sys.argv)
             args = init_args(ResNet, CIFAR10DataModule)
 
             # datamodule