소스 검색

Improved possibility equality test.

Taddeus Kroes 14 년 전
부모
커밋
3909b46f83
1개의 변경된 파일7개의 추가작업 그리고 2개의 파일을 삭제
  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)