test: adjust test cases

iml130 · Jan 27, 2025 · c15555a · c15555a
1 parent 46ae8bd
commit c15555a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/unit_test/model/test_condition.py b/tests/unit_test/model/test_condition.py
@@ -32,7 +32,9 @@ def test_init(self):
         context = ParserRuleContext()
         condition = Condition({"a": 1}, [Service("service")], [TaskCall("task")], context=context)
         self.assertEqual(condition.expression, {"a": 1})
-        self.assertEqual(condition.passed_stmts, [Service("service")])
+        self.assertEqual(1, len(condition.passed_stmts))
+        self.assertIsInstance(condition.passed_stmts[0], Service)
+        self.assertEqual("service", condition.passed_stmts[0].name)
         self.assertEqual(condition.failed_stmts, [TaskCall("task")])
         self.assertEqual(condition.context, context)
         self.assertEqual(condition.context_dict, {})
diff --git a/tests/unit_test/model/test_task.py b/tests/unit_test/model/test_task.py
@@ -42,10 +42,10 @@ def test_init(self):
         )
         self.assertEqual(task.name, "task1")
         self.assertEqual(len(task.statements), 4)
-        self.assertEqual(task.statements[0], Service())
+        self.assertIsInstance(task.statements[0], Service)
         self.assertEqual(task.statements[1], CountingLoop())
         self.assertEqual(task.statements[2], CountingLoop())
-        self.assertEqual(task.statements[3], Service())
+        self.assertIsInstance(task.statements[3], Service)
         self.assertEqual(task.variables, {"var1": "val1", "var2": "val2"})
         self.assertEqual(task.input_parameters, {"in1": "val4", "in2": "val5"})
         self.assertEqual(task.output_parameters, ["out1", "out2"])