Fix tests.

hmcezar · Dec 25, 2024 · e607f4d · e607f4d
1 parent fae1be4
commit e607f4d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/test/conftest.py b/test/conftest.py
@@ -48,6 +48,7 @@ def options_dict(tmp_path):
         "no_hydrogen": True,
         "opt_order": False,
         "solute_natoms": 17,
+        "weight_solute": None,
         "overwrite": True,
         "final_kabsch": False,
         "silhouette_score": False,

diff --git a/test/test_distmat.py b/test/test_distmat.py
@@ -24,6 +24,7 @@ def test_compute_distmat_line(options_dict, clust_opt, first_conf_traj, test_dis
         clust_opt.no_hydrogen,
         None,
         clust_opt.solute_natoms,
+        clust_opt.weight_solute,
         clust_opt.reorder_excl,
         clust_opt.final_kabsch,
     )

diff --git a/test/test_io.py b/test/test_io.py
@@ -65,6 +65,7 @@ def test_extant_file():
 def test_parse_args():
     args = argparse.Namespace(
         natoms_solute=10,
+        weight_solute=None,
         reorder_exclusions=[1, 2, 3],
         reorder_alg="hungarian",
         reorder=False,
@@ -92,6 +93,7 @@ def test_parse_args():
 
     args = argparse.Namespace(
         natoms_solute=10,
+        weight_solute=None,
         reorder_exclusions=[1, 2, 3],
         reorder_alg="hungarian",
         reorder=True,