cache

mlens/utils/tests/test_dummy.py

@@ -14,7 +14,7 @@
 from mlens.utils.dummy import OLS, LogisticRegression, Scale, InitMixin
 from mlens.utils.dummy import ESTIMATORS, PREPROCESSING, ESTIMATORS_PROBA, \
     ECM, ECM_PROBA
-from mlens.utils.dummy import Data, Cache, LayerGenerator
+from mlens.utils.dummy import Data, LayerGenerator
 
 from mlens.utils import assert_correct_format
 from mlens.utils.formatting import _assert_format
@@ -39,13 +39,6 @@
 z[3:8] = 2
 z[8:12] = 3
 
-# Layer estimation
-data = Data('stack', False, False, 2)
-X_, y_ = data.get_data((5, 2), 2)
-_ = data.ground_truth(X_, y_)
-layer = LayerGenerator().get_layer('stack', False, False)
-layer.indexer.fit(X_)
-
 
 if check_estimator is not None:
 
@@ -251,7 +244,6 @@ def test_ground_truth():
                     [-4.        ,  6.28571429],
                     [ 3.14285714,  3.14285714]])
 
-
     t, z = Data('stack', False, True).get_data((6, 2), 2)
 
     with open(os.devnull, 'w') as f, redirect_stdout(f):
@@ -261,10 +253,3 @@ def test_ground_truth():
     np.testing.assert_array_almost_equal(wf, gwf)
     np.testing.assert_array_almost_equal(P, gp)
     np.testing.assert_array_almost_equal(wp, gwp)
-
-
-def test_cache():
-    """[Utils] testing: test cache for layer estimation."""
-    cache = Cache(X_, y_, data)
-    _ = cache.layer_est(layer, 'fit')
-    cache.terminate()