Reduce num_samples for diploid tests

astheeggeggs · Jun 21, 2024 · 451709b · 451709b
1 parent 75b8525
commit 451709b
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_api_fb_diploid.py b/tests/test_api_fb_diploid.py
@@ -66,7 +66,7 @@ def test_ts_simple_n10_no_recomb(self, scale_mutation_rate, include_ancestors):
         ts = self.get_ts_simple_n10_no_recomb()
         self.verify(ts, scale_mutation_rate, include_ancestors)
 
-    @pytest.mark.parametrize("num_samples", [4, 8, 16, 32])
+    @pytest.mark.parametrize("num_samples", [4, 8, 16])
     @pytest.mark.parametrize("scale_mutation_rate", [True, False])
     @pytest.mark.parametrize("include_ancestors", [True, False])
     def test_ts_simple(self, num_samples, scale_mutation_rate, include_ancestors):

diff --git a/tests/test_api_vit_diploid.py b/tests/test_api_vit_diploid.py
@@ -47,7 +47,7 @@ def test_ts_simple_n10_no_recomb(self, scale_mutation_rate, include_ancestors):
         ts = self.get_ts_simple_n10_no_recomb()
         self.verify(ts, scale_mutation_rate, include_ancestors)
 
-    @pytest.mark.parametrize("num_samples", [4, 8, 16, 32])
+    @pytest.mark.parametrize("num_samples", [4, 8, 16])
     @pytest.mark.parametrize("scale_mutation_rate", [True, False])
     @pytest.mark.parametrize("include_ancestors", [True, False])
     def test_ts_simple(self, num_samples, scale_mutation_rate, include_ancestors):

diff --git a/tests/test_nontree_fb_diploid.py b/tests/test_nontree_fb_diploid.py
@@ -87,7 +87,7 @@ def test_ts_simple_n10_no_recomb(
             include_extreme_rates=include_extreme_rates,
         )
 
-    @pytest.mark.parametrize("num_samples", [4, 8, 16, 32])
+    @pytest.mark.parametrize("num_samples", [4, 8, 16])
     @pytest.mark.parametrize("scale_mutation_rate", [True, False])
     @pytest.mark.parametrize("include_ancestors", [True, False])
     @pytest.mark.parametrize("normalise", [True, False])

diff --git a/tests/test_nontree_vit_diploid.py b/tests/test_nontree_vit_diploid.py
@@ -86,7 +86,7 @@ def test_ts_simple_n10_no_recomb(self, scale_mutation_rate, include_ancestors):
         ts = self.get_ts_simple_n10_no_recomb()
         self.verify(ts, scale_mutation_rate, include_ancestors)
 
-    @pytest.mark.parametrize("num_samples", [4, 8, 16, 32])
+    @pytest.mark.parametrize("num_samples", [4, 8, 16])
     @pytest.mark.parametrize("scale_mutation_rate", [True, False])
     @pytest.mark.parametrize("include_ancestors", [True, False])
     def test_ts_simple(self, num_samples, scale_mutation_rate, include_ancestors):