From 5326407d7a787a767dc833a402a71ddf3fda1ed4 Mon Sep 17 00:00:00 2001 From: Gard Spreemann Date: Sun, 28 Feb 2021 22:59:05 +0100 Subject: Don't wait for end of tests to assert success --- src/python/test/test_betti_curve_representations.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/python/test/test_betti_curve_representations.py b/src/python/test/test_betti_curve_representations.py index 1c905b82..5b95fa2c 100755 --- a/src/python/test/test_betti_curve_representations.py +++ b/src/python/test/test_betti_curve_representations.py @@ -9,7 +9,6 @@ def test_betti_curve_is_irregular_betti_curve_followed_by_interpolation(): pinf = 0.05 pzero = 0.05 res = 100 - success = True pds = [] for i in range(0, m): @@ -26,8 +25,8 @@ def test_betti_curve_is_irregular_betti_curve_followed_by_interpolation(): bc2 = BettiCurve2(None) bettis2 = bc2.fit_transform(pds) - success = success and (bc2.grid_ == bc.grid_).all() - success = success and (bettis2 == bettis).all() + assert((bc2.grid_ == bc.grid_).all()) + assert((bettis2 == bettis).all()) for i in range(0, m): grid = np.linspace(pds[i][np.isfinite(pds[i])].min(), pds[i][np.isfinite(pds[i])].max() + 1, res) @@ -37,6 +36,4 @@ def test_betti_curve_is_irregular_betti_curve_followed_by_interpolation(): interp = scipy.interpolate.interp1d(bc.grid_, bettis[i, :], kind="previous", fill_value="extrapolate") bettis_interp = np.array(interp(grid), dtype=int) - success = success and (bettis_interp == bettis_gridded).all() - - assert(success) + assert((bettis_interp == bettis_gridded).all()) -- cgit v1.2.3