class IndexDestroyTest(unittest.TestCase):
def setUp(self) -> None:
self.index = Index(str)
@given(value=text(), index=integers())
@settings(max_examples=100)
def test_destroy(self, value, index):
self.index.add(value, index)
self.index.destroy(value, index)
new_entry = self.index.retrieve(value)
assert new_entry == set()
with pytest.raises(KeyError):
self.index.destroy(value, index)
def test_destroy():
index = Index(bytes)
index.add(None, 0)
index.add(b"23", 1)
index.add(b"23", 2)
index.add(b"23", 3)
index.destroy(None, 0)
assert len(index) == 1
index.destroy(b"23", 1)
assert len(index) == 1
assert index.retrieve(b"23") == {2, 3}
index.destroy(b"23", 2)
assert len(index) == 1
assert index.retrieve(b"23") == {3}
index.destroy(b"23", 3)
assert len(index) == 0
assert index.retrieve(b"23") == set()
with pytest.raises(KeyError):
index.destroy(None, 0)
with pytest.raises(KeyError):
index.destroy(b"23", 1)