def test_get_bucket_sets_value(): db = MockLMDB('vector') sut = LmdbStorage(db) sut.store_vector('hash_name', 'bucket_key', 'v', 'key') buckets = sut.get_bucket('hash_name', 'bucket_key') assert len(buckets) == 1 assert buckets[0] == ('vector', 'key')
def test_store_vector_maps_value(): def duplicate(x): return x + x db = MockLMDB('vector') sut = LmdbStorage(db, duplicate) sut.store_vector('hash_name', 'bucket_key', 'v', 'key') buckets = sut.get_bucket('hash_name', 'bucket_key') assert len(buckets) == 1 assert buckets[0] == ('vectorvector', 'key')
def test_bucket_is_empty(): db = MockLMDB(None) sut = LmdbStorage(db) assert len(sut.get_bucket('hash_name', 'bucket_key')) == 0