ai-content-maker/.venv/Lib/site-packages/pandas/tests/frame/test_npfuncs.py

29 lines
853 B
Python

"""
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_homogenous(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))