Skip to content

Commit 938ab41

Browse files
author
shirzady1934
committed
issue 1162 added test_loc_list_str function for test loc issue with list of str's and instlled pre-commit
1 parent 23f3d35 commit 938ab41

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

tests/test_frame.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3585,17 +3585,19 @@ def test_in_columns() -> None:
35853585
check(assert_type(df[cols], pd.DataFrame), pd.DataFrame)
35863586
check(assert_type(df.groupby(by=cols).sum(), pd.DataFrame), pd.DataFrame)
35873587

3588+
35883589
def test_loc_list_str() -> None:
35893590
# GH 1162 (PR)
3590-
df = pd.DataFrame([[1, 2], [4, 5], [7, 8]],
3591-
index=['cobra', 'viper', 'sidewinder'],
3592-
columns=['max_speed', 'shield'])
3591+
df = pd.DataFrame(
3592+
[[1, 2], [4, 5], [7, 8]],
3593+
index=["cobra", "viper", "sidewinder"],
3594+
columns=["max_speed", "shield"],
3595+
)
35933596

3594-
result = df.loc[['viper', 'sidewinder']]
3597+
result = df.loc[["viper", "sidewinder"]]
35953598
check(assert_type(result, pd.DataFrame), pd.DataFrame)
35963599

35973600

3598-
35993601
def test_insert_newvalues() -> None:
36003602
df = pd.DataFrame({"a": [1, 2]})
36013603
ab = pd.DataFrame({"col1": [1, 2], "col2": [3, 4]})

0 commit comments

Comments
 (0)