Improved possibility equality test.

3909b46f · Taddeus Kroes · e08af8cf · 3909b46f
Commit 3909b46f authored Jan 14, 2012 by Taddeus Kroes
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 2 deletions

tests/rulestestcase.py tests/rulestestcase.py +7 -2

No files found.
--- a/tests/rulestestcase.py
+++ b/tests/rulestestcase.py
@@ -10,6 +10,11 @@ class RulesTestCase(unittest.TestCase):
        for p, e in zip(possibilities, expected):
            self.assertEqual(p.root, e.root)

+            if p.args == None:
+                self.assertIsNone(e.args)
+            elif e.args == None:
+                self.assertIsNone(p.args)
+            else:
                for pair in zip(p.args, e.args):
                    self.assertEqual(*pair)