Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
import sqlite3
import sys
import threading
import time
import warnings
try:
import cPickle as pickle
except:
import pickle
import diskcache as dc
from diskcache.core import ENOVAL
warnings.simplefilter('error')
warnings.simplefilter('ignore', category=dc.EmptyDirWarning)
if sys.hexversion < 0x03000000:
range = xrange
def setup_cache(func):
@ft.wraps(func)
def wrapper():
shutil.rmtree('tmp', ignore_errors=True)
with dc.OrderedCache('tmp') as cache:
func(cache)
shutil.rmtree('tmp', ignore_errors=True)
return wrapper
@setup_cache
def test_init(cache):
import subprocess as sp
import sys
import tempfile
import threading
import time
import warnings
try:
import cPickle as pickle
except:
import pickle
import diskcache as dc
warnings.simplefilter('error')
warnings.simplefilter('ignore', category=dc.EmptyDirWarning)
if sys.hexversion < 0x03000000:
range = xrange
@pytest.fixture
def cache():
with dc.FanoutCache() as cache:
yield cache
shutil.rmtree(cache.directory, ignore_errors=True)
def test_init(cache):
default_settings = dc.DEFAULT_SETTINGS.copy()
del default_settings['size_limit']
for key, value in default_settings.items():
import sys
import tempfile
import threading
import time
import unittest
import warnings
try:
import cPickle as pickle
except:
import pickle
import diskcache
import diskcache as dc
pytestmark = pytest.mark.filterwarnings('ignore', category=dc.EmptyDirWarning)
if sys.hexversion < 0x03000000:
range = xrange
@pytest.fixture
def cache():
with dc.Cache() as cache:
yield cache
shutil.rmtree(cache.directory, ignore_errors=True)
def test_init(cache):
for key, value in dc.DEFAULT_SETTINGS.items():
assert getattr(cache, key) == value
cache.check()
cache.close()
process_queue[index % processes].append(ops)
for num in range(processes):
with open('input-%s.pkl' % num, 'wb') as writer:
pickle.dump(process_queue[num], writer, protocol=2)
for process in subprocs:
process.start()
for process in subprocs:
process.join()
with Cache('tmp') as cache:
warnings.simplefilter('error')
warnings.simplefilter('ignore', category=UnknownFileWarning)
warnings.simplefilter('ignore', category=EmptyDirWarning)
cache.check()
timings = co.defaultdict(list)
for num in range(processes):
with open('output-%s.pkl' % num, 'rb') as reader:
data = pickle.load(reader)
for key in data:
timings[key] += data[key]
if delete:
for num in range(processes):
os.remove('input-%s.pkl' % num)
os.remove('output-%s.pkl' % num)
display(eviction_policy, timings)
process_queue[index % processes].append(ops)
for num in range(processes):
with open('input-%s.pkl' % num, 'wb') as writer:
pickle.dump(process_queue[num], writer, protocol=2)
for process in subprocs:
process.start()
for process in subprocs:
process.join()
with FanoutCache('tmp') as cache:
warnings.simplefilter('error')
warnings.simplefilter('ignore', category=UnknownFileWarning)
warnings.simplefilter('ignore', category=EmptyDirWarning)
cache.check()
timings = {'get': [], 'set': [], 'delete': [], 'self': 0.0}
for num in range(processes):
with open('output-%s.pkl' % num, 'rb') as reader:
data = pickle.load(reader)
for key in data:
timings[key] += data[key]
if delete:
for num in range(processes):
os.remove('input-%s.pkl' % num)
os.remove('output-%s.pkl' % num)
display(eviction_policy, timings)