def test_search_wrong_type(self): uut = SearcherCollection([ RegexSearcher(b'omicron'), RegexSearcher(b'[eu]psilon'), ]) with self.assertRaises(TypeError): uut.search(u('pi omicron mu'))
def test_multi_regex_single_match(self): uut = SearcherCollection([ RegexSearcher('omicron'), RegexSearcher('[eu]psilon'), ]) match = uut.search('pi epsilon iota rho') self.assertIsNotNone(match) self.assertEqual(1, uut.index(match.searcher)) self.assertEqual(3, match.start) self.assertEqual(10, match.end)
def test_constructor(self): s1, s2 = TextSearcher(u('')), TextSearcher(u('')) searcher = SearcherCollection(s1, s2) self.assertEqual(six.text_type, searcher.match_type) self.assertEqual([s1, s2], list(searcher)) searcher = SearcherCollection([s1, s2]) self.assertEqual(six.text_type, searcher.match_type) self.assertEqual([s1, s2], list(searcher)) searcher = SearcherCollection(s1) self.assertEqual([s1], list(searcher)) self.assertEqual(six.text_type, searcher.match_type)
def test_multi_regex_multi_match(self): uut = SearcherCollection([ RegexSearcher(u('omicron')), RegexSearcher(u('[eu]psilon')), TextSearcher(u('pi')), TextSearcher(u('iota')), ]) match = uut.search(u('pi iota epsilon upsilon omicron')) self.assertIsNotNone(match) self.assertEqual(2, uut.index(match.searcher)) self.assertEqual(0, match.start) self.assertEqual(2, match.end)
def test_constructor_invalid(self): with self.assertRaises(ValueError): SearcherCollection([]) with self.assertRaises(TypeError): SearcherCollection(1) with self.assertRaises(ValueError): SearcherCollection(TextSearcher(u('')), BytesSearcher(b'')) NoSearchSearcher = type('NoSearchSearcher', (object,), {'match_type': None}) with self.assertRaises(TypeError): SearcherCollection(NoSearchSearcher()) NoMatchTypeSearcher = type('NoMatchTypeSearcher', (object,), {'search': None}) with self.assertRaises(TypeError): SearcherCollection(NoMatchTypeSearcher())
def test_repr(self): # Only check no exceptions thrown searcher = SearcherCollection([TextSearcher(u('epsilon')), RegexSearcher(u('[eu]psilon'))]) repr(searcher)