Skip to content

Commit 28e2954

Browse files
committed
make test mirror GH pandas-dev#54191
1 parent 3f0e816 commit 28e2954

File tree

1 file changed

+5
-12
lines changed

1 file changed

+5
-12
lines changed

pandas/tests/frame/methods/test_convert_dtypes.py

+5-12
Original file line numberDiff line numberDiff line change
@@ -168,17 +168,10 @@ def test_convert_dtypes_pyarrow_to_np_nullable(self):
168168
expected = pd.DataFrame(range(2), dtype="Int32")
169169
tm.assert_frame_equal(result, expected)
170170

171-
def test_convert_dtypes_timestamp_and_duration(self):
171+
def test_convert_dtypes_pyarrow_timestamp(self):
172172
# GH 54191
173173
pytest.importorskip("pyarrow")
174-
timestamp_series = pd.Series(
175-
pd.date_range("2020-01-01", "2020-01-02", freq="1440min")
176-
).astype("timestamp[ms][pyarrow]")
177-
duration_series = pd.Series(pd.timedelta_range("1s", "2s", freq="1s")).astype(
178-
"duration[ms][pyarrow]"
179-
)
180-
181-
expected = pd.concat([timestamp_series, duration_series], axis=1)
182-
result = expected.convert_dtypes(dtype_backend="pyarrow")
183-
184-
tm.assert_frame_equal(expected, result)
174+
ser = pd.Series(pd.date_range("2020-01-01", "2020-01-02", freq="1min"))
175+
expected = ser.astype("timestamp[ms][pyarrow]")
176+
result = ser.convert_dtypes(dtype_backend="pyarrow")
177+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)