def test_size_limit_constraint(tmpdir): """Testing trivial NullCache: should trigger value generating function on each run""" # We will write total of 5MB to the cache (50KB items x 100) RECORD_SIZE_BYTES = 50 * KB RECORDS_COUNT = 100 a_record = np.random.randint(0, 255, (RECORD_SIZE_BYTES,), np.uint8) cache = LocalDiskCache(tmpdir.strpath, 1 * MB, RECORD_SIZE_BYTES, shards=1) for i in range(RECORDS_COUNT): cache.get('some_key_{}'.format(i), lambda: a_record) # Check that we are more or less within the size limit assert _recursive_folder_size(tmpdir.strpath) < 2 * MB
def test_simple_scalar_cache(tmpdir): """Testing trivial NullCache: should trigger value generating function on each run""" cache = LocalDiskCache(tmpdir.strpath, 1 * MB, 4) assert 42 == cache.get('some_key', lambda: 42) assert 42 == cache.get('some_key', lambda: 43)