Skip to content

Commit f00b7c6

Browse files
Update test_frame.py: used check(assert_type pattern in unit test
1 parent 5807199 commit f00b7c6

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/test_frame.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2348,11 +2348,11 @@ def test_join() -> None:
23482348
def test_types_join() -> None:
23492349
df1 = pd.DataFrame({"A":[1,2], "B": ["test", "test"]})
23502350
df2 = pd.DataFrame({"C":[2, 3], "D": ["test", "test"]})
2351-
df1.join(df2, how="cross")
2352-
df1.join(df2, how="inner")
2353-
df1.join(df2, how="outer")
2354-
df1.join(df2, how="left")
2355-
df1.join(df2, how="right")
2351+
check(assert_type(df1.join(df2, how="cross"), pd.DataFrame), pd.DataFrame)
2352+
check(assert_type(df1.join(df2, how="inner"), pd.DataFrame), pd.DataFrame)
2353+
check(assert_type(df1.join(df2, how="outer"), pd.DataFrame), pd.DataFrame)
2354+
check(assert_type(df1.join(df2, how="left"), pd.DataFrame), pd.DataFrame)
2355+
check(assert_type(df1.join(df2, how="right"), pd.DataFrame), pd.DataFrame)
23562356

23572357

23582358
def test_types_ffill() -> None:

0 commit comments

Comments
 (0)