Merge pull request #7324 from ThomasWaldmann/hashtable-stresstests-master

add hashtable stress tests
This commit is contained in:
TW 2023-02-08 14:48:58 +01:00 committed by GitHub
commit 978cb163e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 37 additions and 0 deletions

View File

@ -0,0 +1,37 @@
import os
import random
import pytest
from ..hashindex import NSIndex
@pytest.mark.skipif("BORG_TESTS_SLOW" not in os.environ, reason="slow tests not enabled, use BORG_TESTS_SLOW=1")
def test_hashindex_stress():
"""checks if the hashtable behaves as expected
This can be used in _hashindex.c before running this test to provoke more collisions (don't forget to compile):
#define HASH_MAX_LOAD .99
#define HASH_MAX_EFF_LOAD .999
"""
ENTRIES = 10000
LOOPS = 1000
idx = NSIndex()
kv = {}
for i in range(LOOPS):
# put some entries
for j in range(ENTRIES):
k = random.randbytes(32)
v = random.randint(0, NSIndex.MAX_VALUE - 1)
idx[k] = (v, v, v)
kv[k] = v
# check and delete a random amount of entries
delete_keys = random.sample(list(kv), k=random.randint(0, len(kv)))
for k in delete_keys:
v = kv.pop(k)
assert idx.pop(k) == (v, v, v)
# check if remaining entries are as expected
for k, v in kv.items():
assert idx[k] == (v, v, v)
# check entry count
assert len(kv) == len(idx)