Add a test

Signed-off-by: Igoshev, Iaroslav <[email protected]>
YarShev · Apr 22, 2024 · 878c295 · 878c295
1 parent 97ea4bf
commit 878c295
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/modin/tests/pandas/dataframe/test_indexing.py b/modin/tests/pandas/dataframe/test_indexing.py
@@ -2176,6 +2176,11 @@ def test___setitem__(data):
 
     df_equals(modin_df, pandas_df)
 
+    # Test df assignment to a columns selection
+    modin_df[modin_df.columns[[0, -1]]] = modin_df[modin_df.columns[[0, -1]]]
+    pandas_df[pandas_df.columns[[0, -1]]] = pandas_df[pandas_df.columns[[0, -1]]]
+    df_equals(modin_df, pandas_df)
+
     # Test series assignment to column
     modin_df = pd.DataFrame(columns=modin_df.columns)
     pandas_df = pandas.DataFrame(columns=pandas_df.columns)