Update to new enum codes, fix tests

Author: dshemetov

safegraph/delphi_safegraph/process.py

@@ -10,7 +10,7 @@
     add_prefix,
     create_export_csv,
     GeoMapper,
-    NAN_CODES,
+    Nans,
 )
 
 from .constants import HOME_DWELL, COMPLETELY_HOME, FULL_TIME_WORK, PART_TIME_WORK, GEO_RESOLUTIONS
@@ -220,19 +220,19 @@ def process_window(df_list: List[pd.DataFrame],
             df_export["timestamp"] = date.strftime('%Y%m%d')
 
             # Default missingness codes
-            df_export["missing_val"] = NAN_CODES["Not Missing"]
-            df_export["missing_se"] = NAN_CODES["Not Missing"]
+            df_export["missing_val"] = Nans.NOT_MISSING
+            df_export["missing_se"] = Nans.NOT_MISSING
             # Sample size will never be missing in this indicator
             # since sample_size just counts the presence of rows for a geo region
-            df_export["missing_sample_size"] = NAN_CODES["Not Missing"]
+            df_export["missing_sample_size"] = Nans.NOT_MISSING
             # Add missingness codes as detected
             # This may occur if all the values are missing for a geographic region
             remaining_nans_mask = df_export["val"].isnull()
-            df_export.loc[remaining_nans_mask, "missing_val"] = NAN_CODES["Unknown"]
+            df_export.loc[remaining_nans_mask, "missing_val"] = Nans.UNKNOWN
             # This may occur if all the values are missing for a geographic region
             # or if the sample size is 1
             remaining_nans_mask = df_export["se"].isnull()
-            df_export.loc[remaining_nans_mask, "missing_se"] = NAN_CODES["Data Insufficient"]
+            df_export.loc[remaining_nans_mask, "missing_se"] = Nans.DATA_INSUFFICIENT
 
             create_export_csv(df_export,
                               export_dir,