diff --git a/tests/test_data.py b/tests/test_data.py index e0bc86d..d194f64 100644 --- a/tests/test_data.py +++ b/tests/test_data.py @@ -11,7 +11,7 @@ from .ori_data_loader import load_colon_data, load_leukemia_data, load_prostate_data -@pytest.mark.flaky(reruns=3, reruns_delay=5) +@pytest.mark.flaky(reruns=5, reruns_delay=30) def test_load_colon_data(): result = _load_colon_data() # only load data not labels assert result is not None @@ -19,7 +19,7 @@ def test_load_colon_data(): assert result.shape == (62, 2000) -@pytest.mark.flaky(reruns=3, reruns_delay=5) +@pytest.mark.flaky(reruns=5, reruns_delay=30) def test_load_colon_label(): result = _load_colon_label() # only load labels not data assert result is not None @@ -27,7 +27,7 @@ def test_load_colon_label(): assert len(result) == 62 -@pytest.mark.flaky(reruns=3, reruns_delay=5) +@pytest.mark.flaky(reruns=5, reruns_delay=30) def test_load_colon(tmpdir): result = load_colon(tmpdir) assert result is not None @@ -39,7 +39,7 @@ def test_load_colon(tmpdir): assert result[1].shape == (62, 2000) -@pytest.mark.flaky(reruns=3, reruns_delay=5) +@pytest.mark.flaky(reruns=5, reruns_delay=30) def test_load_colon_compare_original(tmpdir): result = load_colon(tmpdir) ori_result = load_colon_data() @@ -49,7 +49,7 @@ def test_load_colon_compare_original(tmpdir): assert_almost_equal(result[1].to_numpy(), ori_result[1].to_numpy()) -@pytest.mark.flaky(reruns=3, reruns_delay=5) +@pytest.mark.flaky(reruns=5, reruns_delay=30) def test_load_prostate(tmpdir): result = load_prostate(tmpdir) assert result is not None @@ -61,7 +61,7 @@ def test_load_prostate(tmpdir): assert result[1].shape == (102, 6033) -@pytest.mark.flaky(reruns=3, reruns_delay=5) +@pytest.mark.flaky(reruns=5, reruns_delay=30) def test_load_prostate_compare_original(tmpdir): result = load_prostate(tmpdir) ori_result = load_prostate_data() @@ -71,7 +71,7 @@ def test_load_prostate_compare_original(tmpdir): assert_almost_equal(result[1].to_numpy(), ori_result[1].to_numpy()) -@pytest.mark.flaky(reruns=3, reruns_delay=5) +@pytest.mark.flaky(reruns=5, reruns_delay=30) def test_load_leukemia_big(tmpdir): result = load_leukemia_big(tmpdir) assert result is not None @@ -83,7 +83,7 @@ def test_load_leukemia_big(tmpdir): assert result[1].shape == (72, 7128) -@pytest.mark.flaky(reruns=3, reruns_delay=5) +@pytest.mark.flaky(reruns=5, reruns_delay=30) def test_load_leukemia_big_compare_original(tmpdir): result = load_leukemia_big(tmpdir) ori_result = load_leukemia_data()