def make_pat(): select = any('SELECT', [ "SELECT\\b\\s+\\b", "FROM\\b\\s+\\b", "\\b\\s+WHERE", "[\\b\\s]?\)", "[\\b\\s]?;" ]) update = any('UPDATE', ["UPDATE\\b\\s+\\b", "[\\b\\s]?SET"]) insert = any('INSERT', [ "INSERT\\b\\s+INTO\\b\\s+", "[\\b\\s]?\(", "\\b\\s+SELECT", "\\b\\s+VALUES" ]) return select + '|' + update + '|' + insert
def test_any(self): self.assertTrue(colorizer.any('test', ('a', 'b')))
def test_any(self): self.assertEqual(colorizer.any('test', ('a', 'b', 'cd')), '(?P<test>a|b|cd)')