FUDS needed an extra mock (#1848)

usds · Sep 22, 2022 · 5a0bbb3 · 5a0bbb3
1 parent e7fe4e2
commit 5a0bbb3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/data/data-pipeline/data_pipeline/tests/sources/us_army_fuds/test_etl.py b/data/data-pipeline/data_pipeline/tests/sources/us_army_fuds/test_etl.py
@@ -185,3 +185,10 @@ def test_tracts_without_fuds_not_in_results(self, mock_etl, mock_paths):
             assert len(df[etl.GEOID_TRACT_FIELD_NAME]) == len(
                 self._FIXTURES_SHARED_TRACT_IDS
             )
+
+    def test_tract_id_lengths(self, mock_etl, mock_paths):
+        with mock.patch(
+            "data_pipeline.etl.sources.us_army_fuds.etl.add_tracts_for_geometries",
+            new=_fake_add_tracts_for_geometries,
+        ):
+            return super().test_tract_id_lengths(mock_etl, mock_paths)