update GatingStrategy tests for some missed signature changes

whitews · Oct 10, 2024 · 47f4b1b · 47f4b1b
1 parent 23a61ed
commit 47f4b1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/gating_strategy_tests.py b/tests/gating_strategy_tests.py
@@ -40,7 +40,7 @@ def test_get_parent_gate_is_none(self):
 
     def test_add_transform_non_transform_class(self):
         gs = fk.GatingStrategy()
-        self.assertRaises(TypeError, gs.add_transform, "not a transform class")
+        self.assertRaises(TypeError, gs.add_transform, transform_id="not a transform class", transform=dict())
 
     def test_add_duplicate_transform_id(self):
         gs = fk.GatingStrategy()
@@ -50,7 +50,7 @@ def test_add_duplicate_transform_id(self):
 
     def test_add_matrix_non_matrix_class(self):
         gs = fk.GatingStrategy()
-        self.assertRaises(TypeError, gs.add_comp_matrix, "not a matrix class")
+        self.assertRaises(TypeError, gs.add_comp_matrix, matrix_id="not a matrix class", matrix=dict())
 
     def test_add_duplicate_matrix_id(self):
         gs = fk.GatingStrategy()