示例#1
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])
示例#2
0
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]