Skip to content
Prev Previous commit
Next Next commit
Fix test
  • Loading branch information
phofl committed Jan 7, 2023
commit a4343408aa27a53120ea111fc2d53f9a800de92d
5 changes: 3 additions & 2 deletions pandas/tests/frame/methods/test_align.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
date_range,
)
import pandas._testing as tm
from pandas.core.internals.managers import using_copy_on_write


class TestDataFrameAlign:
Expand Down Expand Up @@ -40,12 +41,12 @@ def test_frame_align_aware(self):
assert new1.index.tz is timezone.utc
assert new2.index.tz is timezone.utc

def test_align_float(self, float_frame, using_copy_on_write, using_array_manager):
def test_align_float(self, float_frame):
af, bf = float_frame.align(float_frame)
assert af._mgr is not float_frame._mgr

af, bf = float_frame.align(float_frame, copy=False)
if using_copy_on_write or using_array_manager:
if using_copy_on_write():
assert af._mgr is not float_frame._mgr
else:
assert af._mgr is float_frame._mgr
Expand Down