Pipeline: Updates national vax (#3917)

SatcherInstitute · Jan 17, 2025 · bb2379d · bb2379d
1 parent 7638bf3
commit bb2379d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/python/tests/datasources/test_cdc_vaccination_national.py b/python/tests/datasources/test_cdc_vaccination_national.py
@@ -43,7 +43,8 @@ def testWriteToBqRace(mock_bq: mock.MagicMock, mock_csv: mock.MagicMock):
 
     expected_df = pd.read_csv(GOLDEN_DATA["race"], dtype={"population_pct": str, "state_fips": str})
 
-    df = mock_bq.call_args_list[0].args[0]
+    df, _, table_name = mock_bq.call_args_list[0][0]
+    assert table_name == "race_national_current"
 
     assert_frame_equal(
         df,
@@ -72,7 +73,8 @@ def testWriteToBqSex(mock_bq: mock.MagicMock, mock_csv: mock.MagicMock):
 
     expected_df = pd.read_csv(GOLDEN_DATA["sex"], dtype={"population_pct": str, "state_fips": str})
 
-    df = mock_bq.call_args_list[1].args[0]
+    df, _, table_name = mock_bq.call_args_list[1][0]
+    assert table_name == "sex_national_current"
 
     assert_frame_equal(
         df,
@@ -101,7 +103,8 @@ def testWriteToBqAge(mock_bq: mock.MagicMock, mock_csv: mock.MagicMock):
 
     expected_df = pd.read_csv(GOLDEN_DATA["age"], dtype={"vaccinated_pop_pct": str, "state_fips": str})
 
-    df = mock_bq.call_args_list[2].args[0]
+    df, _, table_name = mock_bq.call_args_list[2][0]
+    assert table_name == "age_national_current"
 
     assert_frame_equal(
         df,