diff --git a/pandas/tests/copy_view/test_functions.py b/pandas/tests/copy_view/test_functions.py index 56e4b186350f2..3db78063172ba 100644 --- a/pandas/tests/copy_view/test_functions.py +++ b/pandas/tests/copy_view/test_functions.py @@ -3,7 +3,6 @@ from pandas import ( DataFrame, - Index, Series, concat, merge, @@ -310,87 +309,4 @@ def test_merge_copy_keyword(using_copy_on_write, copy): assert np.shares_memory(get_array(df2, "b"), get_array(result, "b")) else: assert not np.shares_memory(get_array(df, "a"), get_array(result, "a")) - assert not np.shares_memory(get_array(df2, "b"), get_array(result, "b")) - - -def test_join_on_key(using_copy_on_write): - df_index = Index(["a", "b", "c"], name="key") - - df1 = DataFrame({"a": [1, 2, 3]}, index=df_index.copy(deep=True)) - df2 = DataFrame({"b": [4, 5, 6]}, index=df_index.copy(deep=True)) - - df1_orig = df1.copy() - df2_orig = df2.copy() - - result = df1.join(df2, on="key") - - if using_copy_on_write: - assert np.shares_memory(get_array(result, "a"), get_array(df1, "a")) - assert np.shares_memory(get_array(result, "b"), get_array(df2, "b")) - assert np.shares_memory(get_array(result.index), get_array(df1.index)) - assert not np.shares_memory(get_array(result.index), get_array(df2.index)) - else: - assert not np.shares_memory(get_array(result, "a"), get_array(df1, "a")) - assert not np.shares_memory(get_array(result, "b"), get_array(df2, "b")) - - result.iloc[0, 0] = 0 - if using_copy_on_write: - assert not np.shares_memory(get_array(result, "a"), get_array(df1, "a")) - assert np.shares_memory(get_array(result, "b"), get_array(df2, "b")) - - result.iloc[0, 1] = 0 - if using_copy_on_write: - assert not np.shares_memory(get_array(result, "b"), get_array(df2, "b")) - - tm.assert_frame_equal(df1, df1_orig) - tm.assert_frame_equal(df2, df2_orig) - - -def test_join_multiple_dataframes_on_key(using_copy_on_write): - df_index = Index(["a", "b", "c"], name="key") - - df1 = DataFrame({"a": [1, 2, 3]}, index=df_index.copy(deep=True)) - dfs_list = [ - DataFrame({"b": [4, 5, 6]}, index=df_index.copy(deep=True)), - DataFrame({"c": [7, 8, 9]}, index=df_index.copy(deep=True)), - ] - - df1_orig = df1.copy() - dfs_list_orig = [df.copy() for df in dfs_list] - - result = df1.join(dfs_list) - - if using_copy_on_write: - assert np.shares_memory(get_array(result, "a"), get_array(df1, "a")) - assert np.shares_memory(get_array(result, "b"), get_array(dfs_list[0], "b")) - assert np.shares_memory(get_array(result, "c"), get_array(dfs_list[1], "c")) - assert np.shares_memory(get_array(result.index), get_array(df1.index)) - assert not np.shares_memory( - get_array(result.index), get_array(dfs_list[0].index) - ) - assert not np.shares_memory( - get_array(result.index), get_array(dfs_list[1].index) - ) - else: - assert not np.shares_memory(get_array(result, "a"), get_array(df1, "a")) - assert not np.shares_memory(get_array(result, "b"), get_array(dfs_list[0], "b")) - assert not np.shares_memory(get_array(result, "c"), get_array(dfs_list[1], "c")) - - result.iloc[0, 0] = 0 - if using_copy_on_write: - assert not np.shares_memory(get_array(result, "a"), get_array(df1, "a")) - assert np.shares_memory(get_array(result, "b"), get_array(dfs_list[0], "b")) - assert np.shares_memory(get_array(result, "c"), get_array(dfs_list[1], "c")) - - result.iloc[0, 1] = 0 - if using_copy_on_write: - assert not np.shares_memory(get_array(result, "b"), get_array(dfs_list[0], "b")) - assert np.shares_memory(get_array(result, "c"), get_array(dfs_list[1], "c")) - - result.iloc[0, 2] = 0 - if using_copy_on_write: - assert not np.shares_memory(get_array(result, "c"), get_array(dfs_list[1], "c")) - - tm.assert_frame_equal(df1, df1_orig) - for df, df_orig in zip(dfs_list, dfs_list_orig): - tm.assert_frame_equal(df, df_orig) + assert not np.shares_memory(get_array(df2, "b"), get_array(result, "b")) \ No newline at end of file