fix: replace np.NAN

VorTECHsa · Dec 9, 2024 · 6de87c1 · 6de87c1
1 parent 26874de
commit 6de87c1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -465,7 +465,7 @@ def test_schema_metrics_are_not_logged_on_write_if_metrics_flag_is_false(self, m
                 CASTING_WARNING_MSG.format("bool_col", "bool", "object"),
             ),
             (
-                pd.DataFrame.from_records([{"id": 1, "foo_name": "A", "bar": 12, "bool_col": True}, {"id": 2, "foo_name": "B", "bar": 12, "bool_col": np.NAN}]),
+                pd.DataFrame.from_records([{"id": 1, "foo_name": "A", "bar": 12, "bool_col": True}, {"id": 2, "foo_name": "B", "bar": 12, "bool_col": np.NaN}]),
                 "bool",
                 CASTING_WARNING_MSG.format("bool_col", "bool", "object"),
             ),
@@ -516,7 +516,7 @@ def test__has_valid_dtypes_does_not_attempt_to_convert_object_type_to_other_type
                 CASTING_WARNING_MSG.format("bool_col", "bool", "object"),
             ),
             (
-                pd.DataFrame.from_records([{"id": 1, "foo_name": "A", "bar": 12, "bool_col": True}, {"id": 2, "foo_name": "B", "bar": 12, "bool_col": np.NAN}]),
+                pd.DataFrame.from_records([{"id": 1, "foo_name": "A", "bar": 12, "bool_col": True}, {"id": 2, "foo_name": "B", "bar": 12, "bool_col": np.NaN}]),
                 "bool",
                 CASTING_WARNING_MSG.format("bool_col", "bool", "object"),
             ),
@@ -578,7 +578,7 @@ def test__has_valid_dtypes_does_not_attempt_to_convert_object_type_to_other_type
                 CASTING_WARNING_MSG.format("bool_col", "bool", "object"),
             ),
             (
-                pd.DataFrame.from_records([{"id": 1, "foo_name": "A", "bar": 12, "bool_col": True}, {"id": 2, "foo_name": "B", "bar": 12, "bool_col": np.NAN}]),
+                pd.DataFrame.from_records([{"id": 1, "foo_name": "A", "bar": 12, "bool_col": True}, {"id": 2, "foo_name": "B", "bar": 12, "bool_col": np.NaN}]),
                 "bool",
                 CASTING_WARNING_MSG.format("bool_col", "bool", "object"),
             ),
@@ -640,7 +640,7 @@ def test__has_valid_dtypes_does_not_attempt_to_convert_object_type_to_other_type
                 CASTING_WARNING_MSG.format("bool_col", "bool", "object"),
             ),
             (
-                pd.DataFrame.from_records([{"id": 1, "foo_name": "A", "bar": 12, "bool_col": True}, {"id": 2, "foo_name": "B", "bar": 12, "bool_col": np.NAN}]),
+                pd.DataFrame.from_records([{"id": 1, "foo_name": "A", "bar": 12, "bool_col": True}, {"id": 2, "foo_name": "B", "bar": 12, "bool_col": np.NaN}]),
                 "bool",
                 CASTING_WARNING_MSG.format("bool_col", "bool", "object"),
             ),