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