File size: 2,751 Bytes
7885a28 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
"""
Tests for np.foo applied to DataFrame, not necessarily ufuncs.
"""
import numpy as np
from pandas import (
Categorical,
DataFrame,
)
import pandas._testing as tm
class TestAsArray:
def test_asarray_homogeneous(self):
df = DataFrame({"A": Categorical([1, 2]), "B": Categorical([1, 2])})
result = np.asarray(df)
# may change from object in the future
expected = np.array([[1, 1], [2, 2]], dtype="object")
tm.assert_numpy_array_equal(result, expected)
def test_np_sqrt(self, float_frame):
with np.errstate(all="ignore"):
result = np.sqrt(float_frame)
assert isinstance(result, type(float_frame))
assert result.index.is_(float_frame.index)
assert result.columns.is_(float_frame.columns)
tm.assert_frame_equal(result, float_frame.apply(np.sqrt))
def test_sum_deprecated_axis_behavior(self):
# GH#52042 deprecated behavior of df.sum(axis=None), which gets
# called when we do np.sum(df)
arr = np.random.default_rng(2).standard_normal((4, 3))
df = DataFrame(arr)
msg = "The behavior of DataFrame.sum with axis=None is deprecated"
with tm.assert_produces_warning(
FutureWarning, match=msg, check_stacklevel=False
):
res = np.sum(df)
with tm.assert_produces_warning(FutureWarning, match=msg):
expected = df.sum(axis=None)
tm.assert_series_equal(res, expected)
def test_np_ravel(self):
# GH26247
arr = np.array(
[
[0.11197053, 0.44361564, -0.92589452],
[0.05883648, -0.00948922, -0.26469934],
]
)
result = np.ravel([DataFrame(batch.reshape(1, 3)) for batch in arr])
expected = np.array(
[
0.11197053,
0.44361564,
-0.92589452,
0.05883648,
-0.00948922,
-0.26469934,
]
)
tm.assert_numpy_array_equal(result, expected)
result = np.ravel(DataFrame(arr[0].reshape(1, 3), columns=["x1", "x2", "x3"]))
expected = np.array([0.11197053, 0.44361564, -0.92589452])
tm.assert_numpy_array_equal(result, expected)
result = np.ravel(
[
DataFrame(batch.reshape(1, 3), columns=["x1", "x2", "x3"])
for batch in arr
]
)
expected = np.array(
[
0.11197053,
0.44361564,
-0.92589452,
0.05883648,
-0.00948922,
-0.26469934,
]
)
tm.assert_numpy_array_equal(result, expected)
|