def test_fsstore(self):
with tempfile.TemporaryDirectory() as tmpdir:
storage = FSStore(tmpdir)
model = FakeModel('val')
model_pickled = pickle.dumps(model)
storage.write(model_pickled, 'for_testing.model')
assert os.path.isfile(os.path.join(
tmpdir, 'for_testing.model')) == storage.exists(
'for_testing.model') == True
with storage.open("for_testing_compressed.model", "wb") as f:
joblib.dump(model, f, compress=True)
assert storage.exists("for_testing_compressed.model")
with open_sesame(
os.path.join(tmpdir, "for_testing_compressed.model"),
"rb") as f:
model_loaded = joblib.load(f)
assert model.val == model_loaded.val
model_loaded = storage.load('for_testing.model')
model_loaded = pickle.loads(model_loaded)
assert model_loaded.val == 'val'
storage.delete('for_testing.model')
assert os.path.isfile(os.path.join(
tmpdir, 'for_testing.model')) == storage.exists(
'for_testing.model') == False
def test_fsstore(self):
with tempfile.TemporaryDirectory() as tmpdir:
storage = FSStore(tmpdir)
model = pickle.dumps(FakeModel('val'))
storage.write(model, 'for_testing.model')
assert os.path.isfile(os.path.join(
tmpdir, 'for_testing.model')) == storage.exists(
'for_testing.model') == True
model_loaded = storage.load('for_testing.model')
model_loaded = pickle.loads(model_loaded)
assert model_loaded.val == 'val'
storage.delete('for_testing.model')
assert os.path.isfile(os.path.join(
tmpdir, 'for_testing.model')) == storage.exists(
'for_testing.model') == False
