from pandas import Index, Series, date_range import pandas._testing as tm class TestGetNumericData: def test_get_numeric_data_preserve_dtype(self): # get the numeric data obj = Series([1, 2, 3]) result = obj._get_numeric_data() tm.assert_series_equal(result, obj) obj = Series([1, "2", 3.0]) result = obj._get_numeric_data() expected = Series([], dtype=object, index=Index([], dtype=object)) tm.assert_series_equal(result, expected) obj = Series([True, False, True]) result = obj._get_numeric_data() tm.assert_series_equal(result, obj) obj = Series(date_range("20130101", periods=3)) result = obj._get_numeric_data() expected = Series([], dtype="M8[ns]", index=Index([], dtype=object)) tm.assert_series_equal(result, expected)