Update tests to reflect removal of old Betti curve class

2 files changed, 1 insertions, 10 deletions

diff --git a/src/python/test/test_betti_curve_representations.py b/src/python/test/test_betti_curve_representations.py
index 475839ee..73f6f34a 100755
--- a/src/python/test/test_betti_curve_representations.py
+++ b/src/python/test/test_betti_curve_representations.py
@@ -1,7 +1,7 @@
 import numpy as np
 import scipy.interpolate
 
-from gudhi.representations.vector_methods import BettiCurve2
+from gudhi.representations.vector_methods import BettiCurve
 
 def test_betti_curve_is_irregular_betti_curve_followed_by_interpolation():
     m = 10
diff --git a/src/python/test/test_representations.py b/src/python/test/test_representations.py
index 43c914f3..86439655 100755
--- a/src/python/test/test_representations.py
+++ b/src/python/test/test_representations.py
@@ -63,12 +63,3 @@ def test_dummy_atol():
             atol_vectoriser.transform(X=[a, b, c])
 
 
-from gudhi.representations.vector_methods import BettiCurve
-
-
-def test_infinity():
-    a = np.array([[1.0, 8.0], [2.0, np.inf], [3.0, 4.0]])
-    c = BettiCurve(20, [0.0, 10.0])(a)
-    assert c[1] == 0
-    assert c[7] == 3
-    assert c[9] == 2