updated test to use synthetic data

Author: ryanraaschCDC

tests/datasets/test_fips_to_name_improved.py

@@ -1,8 +1,13 @@
+from unittest.mock import patch
+
 import pandas as pd
 import pandera.pandas as pa
 import pytest
 from pandera.errors import SchemaError
 
+from cfa.scenarios.dataops.datasets.schemas.fips_to_name import (
+    tf_synth_data as fips_synth_data,
+)
 from cfa.scenarios.dataops.datasets.schemas.fips_to_name_improved import (
     extract_schema,
     load_schema,
@@ -38,8 +43,19 @@ def test_fips_to_name_improved_schemas():
 
 
 def test_fips_to_name_improved_transform():
+    # Create mock FIPS data that would be returned by get_data
+    # This should match the structure expected by the SQL template
+    # and include state abbreviations that might appear in the raw data
+    mock_fips_data = fips_synth_data
+
     extract_df = raw_synth_data
-    tf_df = transform(extract_df)
+
+    # Mock the get_data function to return our mock FIPS data
+    with patch(
+        "cfa.scenarios.dataops.etl.fips_to_name_improved.get_data"
+    ) as mock_get_data:
+        mock_get_data.return_value = mock_fips_data
+        tf_df = transform(extract_df)
 
     # Check if the transformed DataFrame matches the expected schema
     assert isinstance(load_schema(tf_df), pd.DataFrame)