restore skips for faster test runs

moj-analytical-services · Oct 3, 2024 · 4975fa4 · 4975fa4
1 parent 8cc2e52
commit 4975fa4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_cc_random_graphs.py b/tests/test_cc_random_graphs.py
@@ -20,7 +20,7 @@ def test_small_erdos_renyi_graph(execution_number):
     assert (cc_df.values == nx_df.values).all()
 
 
-# @pytest.mark.skip(reason="Slow")
+@pytest.mark.skip(reason="Slow")
 @pytest.mark.parametrize("execution_number", range(10))
 def test_medium_erdos_renyi_graph(execution_number):
     g = generate_random_graph(graph_size=10000)
@@ -31,7 +31,7 @@ def test_medium_erdos_renyi_graph(execution_number):
     assert (cc_df.values == nx_df.values).all()
 
 
-# @pytest.mark.skip(reason="Slow")
+@pytest.mark.skip(reason="Slow")
 @pytest.mark.parametrize("execution_number", range(2))
 def test_large_erdos_renyi_graph(execution_number):
     g = generate_random_graph(graph_size=100000)

diff --git a/tests/test_cluster_at_multiple_thresholds.py b/tests/test_cluster_at_multiple_thresholds.py
@@ -11,13 +11,13 @@
 from .decorator import mark_with_dialects_excluding
 
 
-@pytest.mark.parametrize("graph_size", [100, 500, 1000, 5000])
+@pytest.mark.parametrize("graph_size", [20, 500, 1000, 5000])
 @mark_with_dialects_excluding()
 def test_cluster_at_multiple_thresholds(test_helpers, dialect, graph_size):
     helper = test_helpers[dialect]
     db_api = helper.DatabaseAPI(**helper.db_api_args())
 
-    if dialect == "spark" and graph_size > 100:
+    if dialect == "spark" and graph_size > 20:
         pytest.skip("Skipping large graph sizes for Spark dialect")
 
     G = generate_random_graph(graph_size)