def test_get_items(self): repo = Repository() repo.add_item(0) repo.add_item(1) self.assertEqual(repo.get_item(condition=lambda item: item == 1), 1) uid = repo.add_item(2) self.assertEqual(repo.get_item(uid=uid), 2) repo.add_item(1) self.assertEqual(repo.get_items(), [0, 1, 2, 1]) self.assertEqual(repo.get_items(lambda item: item == 1), [1, 1]) self.assertEqual(repo.get_items(lambda item: item > 2), []) repo.add_item(3) self.assertEqual(repo.get_items(lambda item: item > 2), [3])
def test_get_items(): repo = Repository() repo.add_item(0) repo.add_item(1) assert repo.get_item(lambda item: item == 1) == 1 repo.add_item(1) assert repo.get_items() == [0, 1, 1] assert repo.get_items(lambda item: item == 1) == [1, 1] assert not repo.get_items(lambda item: item > 1) repo.add_item(2) assert repo.get_items(lambda item: item > 1) == [2]