Skip to content

Commit

Permalink
FIX-modin-project#1620: Add test for reported issue
Browse files Browse the repository at this point in the history
Signed-off-by: Vasilij Litvinov <[email protected]>
  • Loading branch information
vnlitvinov committed Dec 17, 2020
1 parent cee481b commit 103e4b5
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions modin/pandas/test/dataframe/test_indexing.py
Original file line number Diff line number Diff line change
Expand Up @@ -411,6 +411,16 @@ def test_loc_assignment(index, columns):
df_equals(md_df, pd_df)


@pytest.mark.parametrize("columns", [["col1", "col2", "col3"]])
def test_loc_iter_assignment(columns):
md_df, pd_df = create_test_dfs(
{col: [idx] for idx, col in enumerate(columns)}, columns=columns
)
md_df.loc[:, columns[:-1]] = md_df.loc[:, columns[:-1]].astype("float")
pd_df.loc[:, columns[:-1]] = pd_df.loc[:, columns[:-1]].astype("float")
df_equals(md_df, pd_df)


@pytest.mark.parametrize("data", test_data_values, ids=test_data_keys)
def test_loc_nested_assignment(data):
modin_df = pd.DataFrame(data)
Expand Down

0 comments on commit 103e4b5

Please sign in to comment.