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])
