test: float conversion error check

agrc · Jul 31, 2023 · fbac445 · fbac445
1 parent d087ac9
commit fbac445
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_transform.py b/tests/test_transform.py
@@ -442,6 +442,18 @@ def test_switch_to_nullable_int_casts_string_field_with_empty_string(self):
 
         tm.assert_frame_equal(retyped_df, test_df)
 
+    def test_switch_to_nullable_int_raises_on_uncastable_text(self):
+        df = pd.DataFrame({
+            'a': [1, 2, np.nan],
+            'b': [1.1, 1.2, 'foo'],
+        })
+
+        with pytest.raises(
+            TypeError,
+            match=re.escape('Cannot convert one or more fields to nullable ints. Check for non-int/non-np.nan values.')
+        ):
+            retyped_df = palletjack.transform.DataCleaning.switch_to_nullable_int(df, ['a', 'b'])
+
 
 class TestFloatFixing: