def setUp(self):
"""
Define one NFA a that accept all strings in the regular language '00*1'
Define a second NFA accepting all string in the regular language '11*0'
"""
fa = NFA('01')
sa0 = fa.new_state(initial=True, name='sa0')
sa1 = fa.new_state(name='sa1')
sa2 = fa.new_state(final=True, name='sa2')
fa.new_edge(sa0, '0', sa1)
fa.new_edge(sa1, '0', sa1)
fa.new_edge(sa1, '1', sa2)
self.assertTrue(fa.test_input('01'), 'String "01" not accepted')
self.assertFalse(fa.test_input('00'), 'String "00" not rejected')
fb = NFA('01')
sb0 = fb.new_state(initial=True, name='sb0')
sb1 = fb.new_state(name='sb1')
sb2 = fb.new_state(final=True, name='sb2')
fb.new_edge(sb0, '1', sb1)
fb.new_edge(sb1, '1', sb1)
fb.new_edge(sb1, '0', sb2)
self.assertTrue(fb.test_input('10'), 'String "10" not accepted')
self.assertFalse(fb.test_input('11'), 'String "11" not rejected')
self.nfa1 = fa
self.nfa2 = fb
def testClosure(self):
nfa = NFA('01')
s0 = nfa.new_state(initial=True)
s1 = nfa.new_state()
s2 = nfa.new_state()
s3 = nfa.new_state(final=True)
s4 = nfa.new_state()
nfa.new_edge(s0, '0', s1)
nfa.new_edge(s0, Epsilon, s2)
nfa.new_edge(s1, '0', s3)
nfa.new_edge(s1, Epsilon, s4)
nfa.new_edge(s2, '1', s2)
nfa.new_edge(s2, Epsilon, s3)
closure = nfa.closure(s0)
self.assertIn(s0, closure, "Closure start state s0 should be in closure")
self.assertNotIn(s1, closure, "State s1 should not be in closure")
self.assertIn(s2, closure, "State s2 not in closure")
self.assertIn(s3, closure, "State s3 not in closure")
self.assertNotIn(s4, closure, "State s4 should not be in closure")
