From dda53bf491900c5be08bf2e10c1db7c5e738f91c Mon Sep 17 00:00:00 2001
From: jdebacker <jason.debacker@gmail.com>
Date: Fri, 31 May 2024 16:17:22 -0400
Subject: [PATCH] set start and end years for imm rates test

---
 tests/test_demographics.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/test_demographics.py b/tests/test_demographics.py
index 9e37327f0..b49b58b27 100644
--- 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