improve test

Author: ntachukwu

pandas/tests/frame/methods/test_set_axis.py

@@ -57,20 +57,21 @@ def test_set_axis_copy(self, obj, using_copy_on_write):
         result = obj.set_axis(new_index, axis=0)
         tm.assert_equal(expected, result)
         assert result is not obj
-        # check we DID make a copy
-        if not using_copy_on_write:
+        if using_copy_on_write:
+            # check we DID NOT make a copy
             if obj.ndim == 1:
-                assert not tm.shares_memory(result, obj)
+                assert tm.shares_memory(result, obj)
             else:
-                assert not any(
+                assert any(
                     tm.shares_memory(result.iloc[:, i], obj.iloc[:, i])
                     for i in range(obj.shape[1])
                 )
         else:
+            # check we DID make a copy
             if obj.ndim == 1:
-                assert tm.shares_memory(result, obj)
+                assert not tm.shares_memory(result, obj)
             else:
-                assert any(
+                assert not any(
                     tm.shares_memory(result.iloc[:, i], obj.iloc[:, i])
                     for i in range(obj.shape[1])
                 )