def test_get_by_tag_nonexistent(self): header_matches = IHeaderMatchList(self._mlist) header_matches.append('header-1', 'pattern-1', tag='tag1') header_matches.append('header-2', 'pattern-') header_matches.append('header-3', 'pattern-2', tag='tag1') match_tag2 = header_matches.get_by_tag('tag2') self.assertEqual(len(list(match_tag2)), 0)
def test_get_by_tag(self): # Test that we can get a list of header_matches with a specifc tag. header_matches = IHeaderMatchList(self._mlist) header_matches.append('header-1', 'pattern-1', tag='tag1') header_matches.append('header-2', 'pattern-') header_matches.append('header-3', 'pattern-2', tag='tag1') self.assertEqual([(match.header, match.position) for match in header_matches.get_by_tag('tag1')], [('header-1', 0), ('header-3', 2)])