test: Silence pandas warning. (#93)

Silence a pandas warning about default dtype of empty DataFrame by explicitly specifying a dtype.
datamole-ai · Aug 21, 2023 · 3bb0b83 · 3bb0b83
1 parent 98b31cf
commit 3bb0b83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_data_type_inference.py b/tests/test_data_type_inference.py
@@ -31,7 +31,7 @@ def test_inference():
         data_types.infer_data_type(pd.Series([1] + list(range(100)))) == data_types.DataType.NUMERIC
     ), "Should be numeric"
     assert (
-        data_types.infer_data_type(pd.Series()) == data_types.DataType.UNKNOWN
+        data_types.infer_data_type(pd.Series(dtype=pd.Float64Dtype)) == data_types.DataType.UNKNOWN
     ), "Should be unknown"
     assert data_types.infer_data_type(
         pd.Series([True, False]) == data_types.DataType.BOOLEAN