Extend geocode utility to actually support the mappings:

- state_x -> state_y where x,y are in {code, id, name}

Author: dshemetov

_delphi_utils_python/delphi_utils/geomap.py

@@ -305,7 +305,9 @@ def add_geocode(
             )
 
         # state codes are all stored in one table
-        if new_code in state_codes:
+        if from_code in state_codes and new_code in state_codes:
+            crosswalk = self._load_crosswalk(from_code="state", to_code="state")
+        elif new_code in state_codes:
             crosswalk = self._load_crosswalk(from_code=from_code, to_code="state")
             crosswalk = crosswalk.rename(
                 columns={from_code: from_col, new_code: new_col}

_delphi_utils_python/tests/test_geomap.py

@@ -382,6 +382,11 @@ def test_add_geocode(self):
         new_data2 = gmpr.add_geocode(new_data, "state_code", "hhs_region_number")
         assert new_data2["hhs_region_number"].unique().size == 2
 
+        # state_name -> state_id
+        new_data = gmpr.add_geocode(self.zip_data, "zip", "state_name")
+        new_data2 = gmpr.add_geocode(new_data, "state_name", "state_id")
+        assert new_data2.shape == (12, 6)
+
         # fips -> nation
         new_data = gmpr.replace_geocode(self.fips_data_5, "fips", "nation")
         assert new_data.equals(