def test_simple(self): r = NFSM("ABC", 3, "ABC") self.assert_no_references(r) self.assertTrue(r.match("ABC")) self.assertFalse(r.match("CBA")) self.assertFalse(r.match("ABCD")) self.assertFalse(r.match("AABC"))
def test_or_peek_and_constraint(self): r = NFSM("AB|BC", 2, "ABC") self.assert_no_references(r) self.assertEqual(set("AB"), r.peek_slot(0)) self.assertEqual(set("BC"), r.peek_slot(1)) r.constrain_slot(0, set("AC")) self.assertEqual(set("A"), r.peek_slot(0)) self.assertEqual(set("B"), r.peek_slot(1))
def test_simple_constraints(self): r = NFSM("...", 3, "ABC") self.assert_no_references(r) self.assertTrue(r.match("AAA")) self.assertTrue(r.match("ABC")) r.constrain_slot(1, set("AB")) self.assertTrue(r.match("AAA")) self.assertTrue(r.match("CBC")) self.assertFalse(r.match("ACA")) self.assertFalse(r.match("BCA")) r.constrain_slot(0, set("C")) self.assertTrue(r.match("CAB")) self.assertFalse(r.match("BAB"))
def test_multi_or(self): r = NFSM("(DI|NS|TH|OM)*", 8, string.ascii_uppercase) self.assert_no_references(r) self.assertTrue(r.match("DIDIDIDI")) self.assertTrue(r.match("DINSTHOM")) self.assertTrue(r.match("OMTHNSDI")) self.assertFalse(r.match("ADINSTHOZ")) self.assertFalse(r.match("ZZZZZZZZ")) r.constrain_slot(0, set("DZ")) self.assertTrue(r.match("DIDIDIDI")) self.assertTrue(r.match("DINSTHOM")) self.assertFalse(r.match("OMTHNSDI")) self.assertFalse(r.match("ZINSTHOM"))
def _compare(self, regex_str, length, alphabet): myr = NFSM(regex_str, length, alphabet) realr = re.compile(regex_str+"$") for s in ("".join(x) for x in product(alphabet, repeat=length)): self.assertEqual(bool(realr.match(s)), myr.match(s), msg=s)
def test_simple_peek(self): r = NFSM("[ABC][AB]", 2, "ABC") self.assert_no_references(r) self.assertEqual(set("ABC"), r.peek_slot(0)) self.assertEqual(set("AB"), r.peek_slot(1))
def test_backref_simple(self): r = NFSM("(.)\\1", 2, "ABC") self.assertTrue(r.match("AA")) self.assertTrue(r.match("CC")) self.assertFalse(r.match("AB")) r.constrain_slot(0, set("AB")) self.assertTrue(r.match("AA")) self.assertTrue(r.match("BB")) self.assertFalse(r.match("CC")) r.constrain_slot(1, set("BC")) self.assertTrue(r.match("BB")) self.assertFalse(r.match("AA")) self.assertFalse(r.match("CC"))
def test_or_star(self): r = NFSM("(RR|HHH)*.?", 10, string.ascii_uppercase) self.assert_no_references(r) self.assertTrue(r.match("RRRRRRRRRR")) self.assertTrue(r.match("RRRRRRHHHA")) self.assertTrue(r.match("RRRRRRHHHR")) self.assertTrue(r.match("RRRRRRHHHH")) self.assertTrue(r.match("HHHHHHRRRR")) self.assertTrue(r.match("RRHHHRRHHH")) self.assertTrue(r.match("HHHRRRRRRZ")) self.assertFalse(r.match("RHHHHHHHHH")) self.assertFalse(r.match("HHHRRRRRR")) self.assertFalse(r.match("HHHHHHHHHRR")) self.assertFalse(r.match("HHRRRRRRRR")) self.assertFalse(r.match("RRRRRRRRRRZ")) self.assertFalse(r.match("RRRRRRRRRRH")) r.constrain_slot(2, set("H")) self.assertTrue(r.match("RRHHHHHHRR")) self.assertTrue(r.match("HHHRRRRHHH")) self.assertTrue(r.match("HHHHHHHHHZ")) self.assertFalse(r.match("RRRRHHHHHH")) self.assertFalse(r.match("RRRRRRHHHZ")) r.constrain_slot(2, set("R")) self.assertFalse(r.match("HHHHHHHHHZ")) self.assertFalse(r.match("RRRRHHHHHH")) self.assertFalse(r.match("RRRRRRHHHH")) self.assertFalse(r.match("HHHHHHRRRR"))
def test_complex(self): r = NFSM("F.*[AO].*[AO].*", 9, string.ascii_uppercase) self.assert_no_references(r) # A few positive matches self.assertTrue(r.match("FBCODEAFG")) self.assertTrue(r.match("FBOCDEAFG")) self.assertTrue(r.match("FBCADEAFG")) self.assertTrue(r.match("FBCADEFOG")) self.assertTrue(r.match("FODEFOGHI")) self.assertTrue(r.match("FBCAAEFOG")) self.assertTrue(r.match("FBCAOEFHG")) # A few negative matches self.assertFalse(r.match("ABCODEAFG")) self.assertFalse(r.match("FBZCDEAFG")) self.assertFalse(r.match("FBCABEZFG")) self.assertFalse(r.match("FZZZZZZZZ")) r.constrain_slot(1, set("AO")) self.assertFalse(r.match("FBCODEAFG")) self.assertFalse(r.match("FBOCDEAFG")) self.assertFalse(r.match("FBCADEAFG")) self.assertFalse(r.match("FBCADEFOG")) self.assertTrue(r.match("FODEFOGHI")) self.assertFalse(r.match("FBCAAEFOG")) self.assertFalse(r.match("FBCAOEFHG")) self.assertFalse(r.match("ABCODEAFG")) self.assertFalse(r.match("FBZCDEAFG")) self.assertFalse(r.match("FBCAOEZFG")) self.assertFalse(r.match("FZZZZZZZZ"))