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)