Skip to content

Commit a29ecc1

Browse files
committed
Fix dt tests to handle DataFrame output
1 parent 3830b3b commit a29ecc1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pandas/tests/series/test_datetime_values.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def get_expected(s, name):
6565
if isinstance(result, np.ndarray):
6666
if is_integer_dtype(result):
6767
result = result.astype("int64")
68-
elif not is_list_like(result):
68+
elif not is_list_like(result) or isinstance(result, pd.DataFrame):
6969
return result
7070
return Series(result, index=s.index, name=s.name)
7171

@@ -74,6 +74,8 @@ def compare(s, name):
7474
b = get_expected(s, prop)
7575
if not (is_list_like(a) and is_list_like(b)):
7676
assert a == b
77+
elif isinstance(a, pd.DataFrame):
78+
tm.assert_frame_equal(a, b)
7779
else:
7880
tm.assert_series_equal(a, b)
7981

0 commit comments

Comments
 (0)