diff --git a/tests/test_api_fb_haploid_multi.py b/tests/test_api_fb_haploid_multi.py index 34858b0..a90f57a 100644 --- a/tests/test_api_fb_haploid_multi.py +++ b/tests/test_api_fb_haploid_multi.py @@ -66,6 +66,8 @@ def test_ts_multiallelic_n10_no_recomb( @pytest.mark.parametrize("num_samples", [6, 8, 16]) @pytest.mark.parametrize("scale_mutation_rate", [True, False]) @pytest.mark.parametrize("include_ancestors", [True, False]) - def test_ts_multiallelic_n16(self, num_samples, scale_mutation_rate, include_ancestors): + def test_ts_multiallelic_n16( + self, num_samples, scale_mutation_rate, include_ancestors + ): ts = self.get_ts_multiallelic(num_samples) self.verify(ts, scale_mutation_rate, include_ancestors) diff --git a/tests/test_api_vit_haploid_multi.py b/tests/test_api_vit_haploid_multi.py index 4a55f14..5020171 100644 --- a/tests/test_api_vit_haploid_multi.py +++ b/tests/test_api_vit_haploid_multi.py @@ -53,6 +53,8 @@ def test_ts_multiallelic_n10_no_recomb( @pytest.mark.parametrize("num_samples", [6, 8, 16]) @pytest.mark.parametrize("scale_mutation_rate", [True, False]) @pytest.mark.parametrize("include_ancestors", [True, False]) - def test_ts_multiallelic_n16(self, num_samples, scale_mutation_rate, include_ancestors): + def test_ts_multiallelic_n16( + self, num_samples, scale_mutation_rate, include_ancestors + ): ts = self.get_ts_multiallelic(num_samples) self.verify(ts, scale_mutation_rate, include_ancestors)