From 7af68be2c217f4c86df75d972f7d86a427f28e89 Mon Sep 17 00:00:00 2001
From: ivan <ivan.rincon76@gmail.com>
Date: Fri, 7 Aug 2020 05:32:01 +0200
Subject: [PATCH] Update optional_data_hash TD3 unittest. See #21

---
 tests/all_chk.py    |  3 +--
 tests/td3_chk_04.py | 19 +++++++++++--------
 2 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/tests/all_chk.py b/tests/all_chk.py
index 9e683df..35d7cd7 100755
--- 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
index 9a69cdc..df240ae 100644
--- 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__':