增加单元测试

.gitignore

@@ -0,0 +1 @@
+__pycache__

tests/test_automata.py

@@ -5,6 +5,17 @@
 
 
 class TestDFA(unittest.TestCase):
+    def setUp(self):
+        # DFA transitions table example
+        self.transitions = {
+            'q0': {'a': 'q1', 'b': None},
+            'q1': {'a': 'q1', 'b': 'q2'},
+            'q2': {'a': 'q1', 'b': 'q3'},
+            'q3': {'a': None, 'b': None}
+        }
+        self.start = 'q0'
+        self.final = ['q3']
+
     def test_DFA(self):
         transitions = {
             'a': {'1': 'a', '0': 'b'},
@@ -43,6 +54,29 @@ def test_DFA(self):
         self.assertEqual(False, DFA(transitions2, start2, final2, "aaabbb"))
         self.assertEqual(True, DFA(transitions2, start2, final2, "baabba"))
 
+    def test_reject_string(self):
+        # Test case where DFA should reject the string
+        string = 'aaa'
+        result = DFA(self.transitions, self.start, self.final, string)
+        self.assertFalse(result)
+
+    def test_invalid_transition(self):
+        # Test case where DFA encounters an invalid transition
+        string = 'b'
+        result = DFA(self.transitions, self.start, self.final, string)
+        self.assertFalse(result)
+
+    def test_empty_string(self):
+        # Test case where the string is empty, start state is not a final state
+        string = ''
+        result = DFA(self.transitions, self.start, self.final, string)
+        self.assertFalse(result)
+
+    def test_empty_string_with_final_start(self):
+        # Test case where the string is empty and start state is a final state
+        start_final = ['q0']
+        result = DFA(self.transitions, self.start, start_final, '')
+        self.assertTrue(result)
 
 if __name__ == '__main__':
     unittest.main()