if pass imm rates, need to infer pop

rickecon · Feb 16, 2024 · 884c3fb · 884c3fb
1 parent 8b522d8
commit 884c3fb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_demographics.py b/tests/test_demographics.py
@@ -225,6 +225,7 @@ def test_custom_series():
         final_data_year=start_year + 1,
         GraphDiag=False,
         imm_rates=imm_rates,
+        infer_pop=True
     )
     assert np.allclose(pop_dict["imm_rates"][0, :], imm_rates[0, E:])