You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
776 B
36 lines
776 B
1 year ago
|
"""
|
||
|
|
Tests for np.foo applied to Series, not necessarily ufuncs.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from pandas import Series
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
|
||
|
|
class TestPtp:
|
||
|
|
def test_ptp(self):
|
||
|
|
# GH#21614
|
||
|
|
N = 1000
|
||
|
|
arr = np.random.default_rng(2).standard_normal(N)
|
||
|
|
ser = Series(arr)
|
||
|
|
assert np.ptp(ser) == np.ptp(arr)
|
||
|
|
|
||
|
|
|
||
|
|
def test_numpy_unique(datetime_series):
|
||
|
|
# it works!
|
||
|
|
np.unique(datetime_series)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("index", [["a", "b", "c", "d", "e"], None])
|
||
|
|
def test_numpy_argwhere(index):
|
||
|
|
# GH#35331
|
||
|
|
|
||
|
|
s = Series(range(5), index=index, dtype=np.int64)
|
||
|
|
|
||
|
|
result = np.argwhere(s > 2).astype(np.int64)
|
||
|
|
expected = np.array([[3], [4]], dtype=np.int64)
|
||
|
|
|
||
|
|
tm.assert_numpy_array_equal(result, expected)
|