.

tests/test_miro.py

@@ -1,7 +1,6 @@
 """
 end-end test for mutual information regulation
 """
-import pytest
 from tests.utils_test import utils_test_algo
 
 
@@ -12,23 +11,3 @@ def test_miro():
     args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=erm \
         --trainer=miro --nname=alexnet"
     utils_test_algo(args)
-
-def test_miro2():
-    """
-    train with MIRO
-    """
-    args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=erm \
-        --trainer=miro --nname=alexnet \
-        --layers2extract_feats _net_invar_feat.net_torchvision.features.1"
-    utils_test_algo(args)
-
-def test_miro3():
-    """
-    train with MIRO
-    """
-    args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=erm \
-        --trainer=miro --nname=alexnet \
-        --layers2extract_feats features"
-    with pytest.raises(RuntimeError):
-        utils_test_algo(args)
-        raise RuntimeError("This is a runtime error")

tests/test_miro2.py

@@ -0,0 +1,14 @@
+"""
+end-end test for mutual information regulation
+"""
+from tests.utils_test import utils_test_algo
+
+
+def test_miro2():
+    """
+    train with MIRO
+    """
+    args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=erm \
+        --trainer=miro --nname=alexnet \
+        --layers2extract_feats _net_invar_feat.net_torchvision.features.1"
+    utils_test_algo(args)

tests/test_miro3.py

@@ -0,0 +1,17 @@
+"""
+end-end test for mutual information regulation
+"""
+import pytest
+from tests.utils_test import utils_test_algo
+
+
+def test_miro3():
+    """
+    train with MIRO
+    """
+    args = "--te_d=caltech --task=mini_vlcs --debug --bs=2 --model=erm \
+        --trainer=miro --nname=alexnet \
+        --layers2extract_feats features"
+    with pytest.raises(RuntimeError):
+        utils_test_algo(args)
+        raise RuntimeError("This is a runtime error")