Skip to content

Commit

Permalink
FIX-modin-project#1620: Add test for .loc[] ordering
Browse files Browse the repository at this point in the history
Signed-off-by: Vasilij Litvinov <vasilij.n.litvinov@intel.com>
  • Loading branch information
vnlitvinov committed Dec 17, 2020
1 parent 852020d commit 92944ed
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions modin/pandas/test/dataframe/test_indexing.py
Original file line number Diff line number Diff line change
Expand Up @@ -436,6 +436,17 @@ def test_loc_iter_assignment(loc_iter_dfs, reverse_order, axis):
df_equals(md_df, pd_df)


@pytest.mark.parametrize("reverse_order", [False, True])
@pytest.mark.parametrize("axis", [0, 1])
def test_loc_order(loc_iter_dfs, reverse_order, axis):
md_df, pd_df = loc_iter_dfs

select = [slice(None), slice(None)]
select[axis] = sorted(pd_df.axes[axis][:-1], reverse=reverse_order)
select = tuple(select)

df_equals(pd_df.loc[select], md_df.loc[select])

@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 92944ed

Please sign in to comment.