38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
import numpy as np
|
|
import pytest
|
|
|
|
from pandas import DataFrame
|
|
import pandas._testing as tm
|
|
|
|
|
|
class TestSwapAxes:
|
|
def test_swapaxes(self):
|
|
df = DataFrame(np.random.default_rng(2).standard_normal((10, 5)))
|
|
msg = "'DataFrame.swapaxes' is deprecated"
|
|
with tm.assert_produces_warning(FutureWarning, match=msg):
|
|
tm.assert_frame_equal(df.T, df.swapaxes(0, 1))
|
|
tm.assert_frame_equal(df.T, df.swapaxes(1, 0))
|
|
|
|
def test_swapaxes_noop(self):
|
|
df = DataFrame(np.random.default_rng(2).standard_normal((10, 5)))
|
|
msg = "'DataFrame.swapaxes' is deprecated"
|
|
with tm.assert_produces_warning(FutureWarning, match=msg):
|
|
tm.assert_frame_equal(df, df.swapaxes(0, 0))
|
|
|
|
def test_swapaxes_invalid_axis(self):
|
|
df = DataFrame(np.random.default_rng(2).standard_normal((10, 5)))
|
|
msg = "'DataFrame.swapaxes' is deprecated"
|
|
with tm.assert_produces_warning(FutureWarning, match=msg):
|
|
msg = "No axis named 2 for object type DataFrame"
|
|
with pytest.raises(ValueError, match=msg):
|
|
df.swapaxes(2, 5)
|
|
|
|
def test_round_empty_not_input(self):
|
|
# GH#51032
|
|
df = DataFrame({"a": [1, 2]})
|
|
msg = "'DataFrame.swapaxes' is deprecated"
|
|
with tm.assert_produces_warning(FutureWarning, match=msg):
|
|
result = df.swapaxes("index", "index")
|
|
tm.assert_frame_equal(df, result)
|
|
assert df is not result
|