Fix test for Filtering

giotto-ai · Jul 22, 2020 · f8588dc · f8588dc
1 parent 6017caf
commit f8588dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gtda/diagrams/tests/test_preprocessing.py b/gtda/diagrams/tests/test_preprocessing.py
@@ -277,7 +277,7 @@ def test_filt_transform_zero(X):
 def test_filt_transform(epsilon):
     filt = Filtering(epsilon=epsilon)
     X_res_1 = filt.fit_transform(X_1)
-    assert X_res_1.shape == X_1.shape
+    assert X_res_1.shape[1] <= X_1.shape[1]
 
     lifetimes_res_1 = X_res_1[:, :, 1] - X_res_1[:, :, 0]
     assert not ((lifetimes_res_1 > 0.) & (lifetimes_res_1 <= epsilon)).any()