Skip to content

Commit 28b5450

Browse files
committed
added test for groupby
1 parent 6a06a31 commit 28b5450

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

pandas/tests/groupby/test_grouping.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1180,3 +1180,19 @@ def test_grouping_by_key_is_in_axis():
11801180
result = gb.sum()
11811181
expected = DataFrame({"a": [1, 2], "b": [1, 2], "c": [7, 5]})
11821182
tm.assert_frame_equal(result, expected)
1183+
1184+
1185+
def test_groupby_any_with_timedelta(self):
1186+
# Create a DataFrame with Timedelta and NaT values
1187+
df = DataFrame({
1188+
"A": ["foo", "foo", "bar", "bar"],
1189+
"B": [pd.Timedelta(1, unit='D'), pd.NaT, pd.Timedelta(2, unit='D'), pd.NaT]
1190+
})
1191+
1192+
# Group by column A and check if any Timedelta exists (i.e., non-NaT)
1193+
result = df.groupby("A")["B"].any()
1194+
1195+
# Expected result: groups with only NaT should return False, others should return True
1196+
expected = Series([True, False], index=["foo", "bar"], name="B")
1197+
1198+
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)