Explorar o código

Improved possibility equality test.

Taddeus Kroes %!s(int64=14) %!d(string=hai) anos
pai
achega
3909b46f83
Modificáronse 1 ficheiros con 7 adicións e 2 borrados
  1. 7 2
      tests/rulestestcase.py

+ 7 - 2
tests/rulestestcase.py

@@ -10,8 +10,13 @@ class RulesTestCase(unittest.TestCase):
         for p, e in zip(possibilities, expected):
             self.assertEqual(p.root, e.root)
 
-            for pair in zip(p.args, e.args):
-                self.assertEqual(*pair)
+            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)
 
             self.assertEqual(p, e)