Skip to content

Commit 81b1598

Browse files
authored
CLN: Split test_fields.py (pandas-dev#44816)
1 parent 41680b1 commit 81b1598

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

pandas/tests/tslibs/test_fields.py

+14-3
Original file line numberDiff line numberDiff line change
@@ -1,28 +1,39 @@
11
import numpy as np
2+
import pytest
23

34
from pandas._libs.tslibs import fields
45

56
import pandas._testing as tm
67

78

8-
def test_fields_readonly():
9-
# https://github.com/vaexio/vaex/issues/357
10-
# fields functions shouldn't raise when we pass read-only data
9+
@pytest.fixture
10+
def dtindex():
1111
dtindex = np.arange(5, dtype=np.int64) * 10 ** 9 * 3600 * 24 * 32
1212
dtindex.flags.writeable = False
13+
return dtindex
1314

15+
16+
def test_get_date_name_field_readonly(dtindex):
17+
# https://github.com/vaexio/vaex/issues/357
18+
# fields functions shouldn't raise when we pass read-only data
1419
result = fields.get_date_name_field(dtindex, "month_name")
1520
expected = np.array(["January", "February", "March", "April", "May"], dtype=object)
1621
tm.assert_numpy_array_equal(result, expected)
1722

23+
24+
def test_get_date_field_readonly(dtindex):
1825
result = fields.get_date_field(dtindex, "Y")
1926
expected = np.array([1970, 1970, 1970, 1970, 1970], dtype=np.int32)
2027
tm.assert_numpy_array_equal(result, expected)
2128

29+
30+
def test_get_start_end_field_readonly(dtindex):
2231
result = fields.get_start_end_field(dtindex, "is_month_start", None)
2332
expected = np.array([True, False, False, False, False], dtype=np.bool_)
2433
tm.assert_numpy_array_equal(result, expected)
2534

35+
36+
def test_get_timedelta_field_readonly(dtindex):
2637
# treat dtindex as timedeltas for this next one
2738
result = fields.get_timedelta_field(dtindex, "days")
2839
expected = np.arange(5, dtype=np.int32) * 32

0 commit comments

Comments
 (0)