set start and end years for imm rates test

jdebacker · May 31, 2024 · dda53bf · dda53bf
1 parent 6642061
commit dda53bf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_demographics.py b/tests/test_demographics.py
@@ -52,6 +52,8 @@ def test_get_pop_objs_read_UN_data():
         GraphDiag=False,
     )
 
+    assert isinstance(pop_dict, dict)
+
 
 def test_get_pop_objs():
     """
@@ -286,6 +288,8 @@ def test_get_imm_rates():
         mort_rates=mort_rates,
         infmort_rates=infmort_rates,
         pop_dist=pop_dist,
+        start_year=2024,
+        end_year=2025,
         graph=True,
     )
     assert imm_rates.shape[1] == S