Update test for OED4 - test_field_required_allow_blank_are_set_to_def…

…ault (#134)
OasisLMF · Aug 5, 2024 · bdd721c · bdd721c
1 parent 90c8262
commit bdd721c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_ods_package.py b/tests/test_ods_package.py
@@ -411,12 +411,16 @@ def test_field_required_allow_blank_are_set_to_default(self):
             'use_field': True})
 
         original_exposure.location.dataframe.drop(columns=['ContentsTIV'], inplace=True)
+        original_exposure.location.dataframe['ContentsTIV'] = pd.NA
+
         original_exposure.location.dataframe['BITIV'] = pd.NA
         original_exposure.location.dataframe.loc[[1], 'BITIV'] = np.nan
         original_exposure.location.dataframe.loc[[2], 'BITIV'] = None
         original_exposure.location.dataframe.loc[[3], 'BITIV'] = pd.NaT
         original_exposure.location.dataframe.loc[[4], 'BITIV'] = ''
+
         original_exposure.ri_info.dataframe.drop(columns='RiskLevel', inplace=True)
+        original_exposure.ri_info.dataframe['RiskLevel'] = pd.NA
 
         with tempfile.TemporaryDirectory() as tmp_dir:
             original_exposure.save(pathlib.Path(tmp_dir, 'oed'), save_config=True)