def test_filter_retroencabulator_goldfish(self):
goldfish_filter = dec16.filter_factory('goldfish', 9, True)
aunts = [dec16.Aunt(line) for line in self.aunt_lines]
filtered_aunts = list(filter(goldfish_filter, aunts))
self.assertEquals(2, len(filtered_aunts))
self.assertEquals(27, filtered_aunts[0].number)
self.assertEquals(28, filtered_aunts[1].number)
def test_filter_missing_data(self):
akita_filter = dec16.filter_factory('akitas', 9, False)
aunts = [dec16.Aunt(line) for line in self.aunt_lines]
filtered_aunts = list(filter(akita_filter, aunts))
self.assertEquals(2, len(filtered_aunts))
def test_filter_retroencabulator_trees(self):
tree_filter = dec16.filter_factory('trees', 2, True)
aunts = [dec16.Aunt(line) for line in self.aunt_lines]
filtered_aunts = list(filter(tree_filter, aunts))
self.assertEquals(2, len(filtered_aunts))
def test_filter(self):
tree_filter = dec16.filter_factory('trees', 3, False)
aunts = [dec16.Aunt(line) for line in self.aunt_lines]
filtered_aunts = list(filter(tree_filter, aunts))
self.assertEquals(1, len(filtered_aunts))
self.assertEquals(27, filtered_aunts[0].number)
