FIX-modin-project#2442: axis alignment fixed in case of 'setitem'

Signed-off-by: Dmitry Chigarev <[email protected]>
dchigarev · Dec 9, 2020 · ff5ec3b · ff5ec3b
1 parent db173b2
commit ff5ec3b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/modin/backends/pandas/query_compiler.py b/modin/backends/pandas/query_compiler.py
@@ -2214,10 +2214,10 @@ def setitem(self, axis, key, value):
         Returns:
              A new QueryCompiler
         """
-        if isinstance(value, type(self)) and not value.get_axis(axis).equals(
+        if isinstance(value, type(self)) and not value.get_axis(0).equals(
             self.get_axis(axis)
         ):
-            value = value.reindex(axis, self.get_axis(axis))
+            value = value.reindex(0, self.get_axis(axis))
 
         return self._setitem(axis=axis, key=key, value=value)
 

diff --git a/modin/pandas/test/dataframe/test_indexing.py b/modin/pandas/test/dataframe/test_indexing.py
@@ -399,8 +399,8 @@ def test_loc_multi_index():
     df_equals(modin_df.loc[modin_df.index[:7]], pandas_df.loc[pandas_df.index[:7]])
 
 
-@pytest.mark.parametrize("index", [["row1", "row2", "row3"], ["row1"]])
-@pytest.mark.parametrize("columns", [["col1", "col2"], ["col1"]])
+@pytest.mark.parametrize("index", [["row1", "row2", "row3"]])
+@pytest.mark.parametrize("columns", [["col1", "col2"]])
 def test_loc_assignment(index, columns):
     md_df, pd_df = create_test_dfs(index=index, columns=columns)
     for i, ind in enumerate(index):