try something to patch test ssda

aramis-lab · Oct 18, 2024 · 12b1463 · 12b1463
1 parent c641d23
commit 12b1463
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 0 deletions.
diff --git a/tests/test_train_ae.py b/tests/test_train_ae.py
@@ -106,6 +106,7 @@ def test_train_ae(cmdopt, tmp_path, test_name):
             base_dir=base_dir,
             no_gpu=cmdopt["no-gpu"],
             adapt_base_dir=cmdopt["adapt-base-dir"],
+            ssda=True,
         )
     assert json_data_out == json_data_ref  # ["mode"] == mode
 

diff --git a/tests/test_train_cnn.py b/tests/test_train_cnn.py
@@ -125,6 +125,7 @@ def test_train_cnn(cmdopt, tmp_path, test_name):
             base_dir=base_dir,
             no_gpu=cmdopt["no-gpu"],
             adapt_base_dir=cmdopt["adapt-base-dir"],
+            ssda=True,
         )
     assert json_data_out == json_data_ref  # ["mode"] == mode
 

diff --git a/tests/test_transfer_learning.py b/tests/test_transfer_learning.py
@@ -169,6 +169,7 @@ def test_transfer_learning(cmdopt, tmp_path, test_name):
     json_data_ref["gpu"] = json_data_out["gpu"]
     json_data_ref["transfer_path"] = json_data_out["transfer_path"]
     json_data_ref["tsv_path"] = json_data_out["tsv_path"]
+    json_data_out["ssda_network"] = json_data_ref["ssda_network"]
     ###
     assert json_data_out == json_data_ref  # ["mode"] == mode
 

diff --git a/tests/testing_tools.py b/tests/testing_tools.py
@@ -175,6 +175,7 @@ def modify_maps(
     no_gpu: bool = False,
     adapt_base_dir: bool = False,
     modify_split: bool = False,
+    ssda: bool = False,
 ) -> Dict[str, Any]:
     """
     Modifies a MAPS dictionary if the user passed --no-gpu or --adapt-base-dir flags.
@@ -212,6 +213,9 @@ def modify_maps(
 
     if modify_split:
         maps["split"] = (0,)
+
+    if ssda:
+        maps["ssda_network"] = False
     return maps