def test_not_fitted(): dst = ForgetDimension() dsc = Scaler() with pytest.raises(NotFittedError): dst.transform(X_1) with pytest.raises(NotFittedError): dsc.transform(X_1) with pytest.raises(NotFittedError): dsc.inverse_transform(X_1)
def test_sc_transform_shape(X, metric, metric_params): sc = Scaler(metric=metric, metric_params=metric_params, n_jobs=1) X_res = sc.fit_transform(X) assert X_res.shape == X.shape X_inv_res = sc.inverse_transform(X_res) assert_almost_equal(X_inv_res, X)