Update optional_data_hash TD3 unittest. See #21

Arg0s1080 · Aug 7, 2020 · 7af68be · 7af68be
1 parent 5e812c4
commit 7af68be
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/tests/all_chk.py b/tests/all_chk.py
@@ -29,8 +29,7 @@ def test_checker(self):
         TestCase09().test_c_td3_checker()
         TestCase10().test_td3_checker()
         TestCase17().test_td3_checker()
-        TestCase20().test_a_td3_checker()
-        TestCase20().test_b_td3_checker()
+        TestCase20().test_td3_checker()
         TestCase11().test_mrva_checker()
         TestCase12().test_mrva_checker()
         TestCase18().test_mrva_checker()

diff --git a/tests/td3_chk_04.py b/tests/td3_chk_04.py
@@ -7,15 +7,18 @@ class TestCase20(unittest.TestCase):
     # if optional data is empty "0" and "<" will return True
     # See #21
 
-    def test_a_td3_checker(self):
-        mrz_td3 = ("P<CANMARTIN<<SARAH<<<<<<<<<<<<<<<<<<<<<<<<<<\n"
-                   "ZE000509<9CAN8501019F2301147<<<<<<<<<<<<<<08")
-        self.assertTrue(bool(TD3CodeChecker(mrz_td3)))
+    def test_td3_checker(self):
+        # optional data hash == '0'
+        mrz1 = ("P<CANMARTIN<<SARAH<<<<<<<<<<<<<<<<<<<<<<<<<<\n"
+                "ZE000509<9CAN8501019F2301147<<<<<<<<<<<<<<08")
+        test1 = bool(TD3CodeChecker(mrz1))
 
-    def test_b_td3_checker(self):
-        mrz_td3 = ("P<CANMARTIN<<SARAH<<<<<<<<<<<<<<<<<<<<<<<<<<\n"
-                   "ZE000509<9CAN8501019F2301147<<<<<<<<<<<<<<<8")
-        self.assertTrue(bool(TD3CodeChecker(mrz_td3)))
+        # optional data hash == '<'
+        mrz2 = ("P<CANMARTIN<<SARAH<<<<<<<<<<<<<<<<<<<<<<<<<<\n"
+                "ZE000509<9CAN8501019F2301147<<<<<<<<<<<<<<08")
+        test2 = bool(TD3CodeChecker(mrz2))
+
+        self.assertEqual(test1, test2)
 
 
 if __name__ == '__main__':