Updated a check_args method.

Now the name of an argument that causes assertion failure should be obtainable.
PanDAWMS · Nov 19, 2018 · e1343c5 · mgolosova · Nov 20, 2018 · Evildoor
1 parent 4c1c44a
commit e1343c5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Utils/Dataflow/pyDKB/dataflow/stage/tests/test_ProcessorStage.py b/Utils/Dataflow/pyDKB/dataflow/stage/tests/test_ProcessorStage.py
@@ -48,7 +48,13 @@ def tearDown(self):
 
     def check_args(self, args):
         for a in args:
-            self.assertEqual(getattr(self.stage.ARGS, a), args[a])
+            # Such kind of testing does not display the argument's name,
+            # hence the "a + '_'" addition.
+            if isinstance(args[a], str):
+                self.assertEqual(a + '_' + getattr(self.stage.ARGS, a),
+                                 a + '_' + args[a])
+            else:
+                self.assertEqual(getattr(self.stage.ARGS, a), args[a])
 
     def test_default(self):
         self.stage.parse_args('')