You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
943 B
40 lines
943 B
import pytest
|
|
import random
|
|
|
|
from disco.util.hashmap import HashMap
|
|
|
|
|
|
@pytest.fixture
|
|
def hashmap():
|
|
return HashMap({i: random.randint(1, 1000000) for i in range(100000)})
|
|
|
|
|
|
def test_hashmap_insert_performance(benchmark):
|
|
def bench_hashmap_insert(hsh):
|
|
hsh[random.randint(1, 100000)] = True
|
|
|
|
benchmark(bench_hashmap_insert, HashMap())
|
|
|
|
|
|
def test_hashmap_lookup_performance(benchmark, hashmap):
|
|
def bench_hashmap_lookup():
|
|
assert hashmap[random.randint(1, 10000)] > 0
|
|
|
|
benchmark(bench_hashmap_lookup)
|
|
|
|
|
|
def test_hashmap_find(hashmap):
|
|
assert len(list(hashmap.find(lambda v: v > 0))) == len(hashmap)
|
|
assert hashmap.find_one(lambda v: v > 0) > 0
|
|
|
|
|
|
def test_hashmap_filter(hashmap):
|
|
for item in list(hashmap.filter(lambda v: v % 2 == 0)):
|
|
assert item % 2 == 0
|
|
|
|
|
|
def test_hashmap_builtins(hashmap):
|
|
for item in hashmap:
|
|
assert item in hashmap
|
|
|
|
assert hashmap.popitem()[1] > 1
|
|
|