Please accept this humble offering, Lord Codecov

popsim-consortium · Sep 17, 2022 · 9f7ac79 · 9f7ac79
1 parent 1eae8e7
commit 9f7ac79
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -307,6 +307,8 @@ def test_get_sample_sets(self):
         # Check for error when prohibited sampling asked for
         with pytest.raises(ValueError, match="non-sampling population"):
             base_mod.get_sample_sets({"pop0": 2, "pop1": 2, "pop2": 1}, ploidy=1)
+        # OK if prohibited population has 0 requested samples
+        base_mod.get_sample_sets({"pop0": 2, "pop1": 2, "pop2": 0}, ploidy=1)
         # Check for error with nonexistant population
         with pytest.raises(ValueError, match="is not one of the populations"):
             base_mod.get_sample_sets({"pop0": 2, "pop1": 2, "nonexistant": 1}, ploidy=1)