def test_remove_items(): repo = Repository() repo.add_item(0) repo.add_item(1) repo.add_item(1) repo.add_item(1) repo.remove_item(lambda item: item == 1) assert repo.get_items() == [0, 1, 1] repo.remove_items(lambda item: item == 1) assert repo.get_items() == [0]
def test_replace_item(self): repo = Repository() repo.add_item(0) repo.add_item(1) repo.add_item(1) uid = repo.add_item(1) repo.replace_item(0, condition=lambda item: item == 1) self.assertEqual(repo.get_items(), [0, 0, 1, 1]) repo.replace_item(2, uid=uid) self.assertEqual(repo.get_items(), [0, 0, 1, 2])
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]
def test_remove_items(self): repo = Repository() repo.add_item(0) repo.add_item(1) repo.add_item(1) repo.add_item(1) repo.remove_item(condition=lambda item: item == 1) self.assertEqual(repo.get_items(), [0, 1, 1]) uid = repo.add_item(2) repo.remove_item(uid=uid) self.assertEqual(repo.get_items(), [0, 1, 1]) repo.remove_items(lambda item: item == 1) self.assertEqual(repo.get_items(), [0])
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_replace_item(): repo = Repository() repo.add_item(0) repo.add_item(1) repo.add_item(1) repo.add_item(1) repo.replace_item(lambda item: item == 1, 0) assert repo.get_items() == [0, 0, 1, 1]
def test_add_item(): repo = Repository() repo.add_item(1) assert repo.get_items() == [1]
def test_add_item(self): repo = Repository() repo.add_item(1) self.assertEqual(repo.get_items(), [1])