Skip to content

Commit 80120b7

Browse files
chean.wei.khorchean.wei.khor
chean.wei.khor
authored and
chean.wei.khor
committed
pre commit
1 parent b7c833c commit 80120b7

File tree

7 files changed

+30
-28
lines changed

7 files changed

+30
-28
lines changed

pandas/tests/indexes/interval/test_astype.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ class TestFloatSubtype(AstypeTests):
133133
"""Tests specific to IntervalIndex with float subtype"""
134134

135135
indexes = [
136-
interval_range(-10.0, 10.0, closed="neither"),
136+
interval_range(-10.0, 10.0, inclusive="neither"),
137137
IntervalIndex.from_arrays(
138138
[-1.5, np.nan, 0.0, 0.0, 1.5], [-0.5, np.nan, 1.0, 1.0, 3.0], closed="both"
139139
),
@@ -161,6 +161,7 @@ def test_subtype_integer(self, subtype):
161161
@pytest.mark.parametrize("subtype", ["int64", "uint64"])
162162
def test_subtype_integer_with_non_integer_borders(self, subtype):
163163
index = interval_range(0.0, 3.0, freq=0.25)
164+
164165
dtype = IntervalDtype(subtype, "right")
165166
result = index.astype(dtype)
166167
expected = IntervalIndex.from_arrays(
@@ -191,10 +192,10 @@ class TestDatetimelikeSubtype(AstypeTests):
191192
"""Tests specific to IntervalIndex with datetime-like subtype"""
192193

193194
indexes = [
194-
interval_range(Timestamp("2018-01-01"), periods=10, closed="neither"),
195+
interval_range(Timestamp("2018-01-01"), periods=10, inclusive="neither"),
195196
interval_range(Timestamp("2018-01-01"), periods=10).insert(2, NaT),
196197
interval_range(Timestamp("2018-01-01", tz="US/Eastern"), periods=10),
197-
interval_range(Timedelta("0 days"), periods=10, closed="both"),
198+
interval_range(Timedelta("0 days"), periods=10, inclusive="both"),
198199
interval_range(Timedelta("0 days"), periods=10).insert(2, NaT),
199200
]
200201

pandas/tests/indexes/interval/test_interval.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -167,10 +167,10 @@ def test_delete(self, closed):
167167
@pytest.mark.parametrize(
168168
"data",
169169
[
170-
interval_range(0, periods=10, closed="neither"),
171-
interval_range(1.7, periods=8, freq=2.5, closed="both"),
172-
interval_range(Timestamp("20170101"), periods=12, closed="left"),
173-
interval_range(Timedelta("1 day"), periods=6, closed="right"),
170+
interval_range(0, periods=10, inclusive="neither"),
171+
interval_range(1.7, periods=8, freq=2.5, inclusive="both"),
172+
interval_range(Timestamp("20170101"), periods=12, inclusive="left"),
173+
interval_range(Timedelta("1 day"), periods=6, inclusive="right"),
174174
],
175175
)
176176
def test_insert(self, data):
@@ -868,9 +868,9 @@ def test_nbytes(self):
868868
@pytest.mark.parametrize("new_closed", ["left", "right", "both", "neither"])
869869
def test_set_closed(self, name, closed, new_closed):
870870
# GH 21670
871-
index = interval_range(0, 5, closed=closed, name=name)
871+
index = interval_range(0, 5, inclusive=closed, name=name)
872872
result = index.set_closed(new_closed)
873-
expected = interval_range(0, 5, closed=new_closed, name=name)
873+
expected = interval_range(0, 5, inclusive=new_closed, name=name)
874874
tm.assert_index_equal(result, expected)
875875

876876
@pytest.mark.parametrize("bad_closed", ["foo", 10, "LEFT", True, False])

pandas/tests/indexes/interval/test_interval_range.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -34,25 +34,25 @@ def test_constructor_numeric(self, closed, name, freq, periods):
3434

3535
# defined from start/end/freq
3636
result = interval_range(
37-
start=start, end=end, freq=freq, name=name, closed=closed
37+
start=start, end=end, freq=freq, name=name, inclusive=closed
3838
)
3939
tm.assert_index_equal(result, expected)
4040

4141
# defined from start/periods/freq
4242
result = interval_range(
43-
start=start, periods=periods, freq=freq, name=name, closed=closed
43+
start=start, periods=periods, freq=freq, name=name, inclusive=closed
4444
)
4545
tm.assert_index_equal(result, expected)
4646

4747
# defined from end/periods/freq
4848
result = interval_range(
49-
end=end, periods=periods, freq=freq, name=name, closed=closed
49+
end=end, periods=periods, freq=freq, name=name, inclusive=closed
5050
)
5151
tm.assert_index_equal(result, expected)
5252

5353
# GH 20976: linspace behavior defined from start/end/periods
5454
result = interval_range(
55-
start=start, end=end, periods=periods, name=name, closed=closed
55+
start=start, end=end, periods=periods, name=name, inclusive=closed
5656
)
5757
tm.assert_index_equal(result, expected)
5858

@@ -67,19 +67,19 @@ def test_constructor_timestamp(self, closed, name, freq, periods, tz):
6767

6868
# defined from start/end/freq
6969
result = interval_range(
70-
start=start, end=end, freq=freq, name=name, closed=closed
70+
start=start, end=end, freq=freq, name=name, inclusive=closed
7171
)
7272
tm.assert_index_equal(result, expected)
7373

7474
# defined from start/periods/freq
7575
result = interval_range(
76-
start=start, periods=periods, freq=freq, name=name, closed=closed
76+
start=start, periods=periods, freq=freq, name=name, inclusive=closed
7777
)
7878
tm.assert_index_equal(result, expected)
7979

8080
# defined from end/periods/freq
8181
result = interval_range(
82-
end=end, periods=periods, freq=freq, name=name, closed=closed
82+
end=end, periods=periods, freq=freq, name=name, inclusive=closed
8383
)
8484
tm.assert_index_equal(result, expected)
8585

@@ -88,7 +88,7 @@ def test_constructor_timestamp(self, closed, name, freq, periods, tz):
8888
# matches expected only for non-anchored offsets and tz naive
8989
# (anchored/DST transitions cause unequal spacing in expected)
9090
result = interval_range(
91-
start=start, end=end, periods=periods, name=name, closed=closed
91+
start=start, end=end, periods=periods, name=name, inclusive=closed
9292
)
9393
tm.assert_index_equal(result, expected)
9494

@@ -102,25 +102,25 @@ def test_constructor_timedelta(self, closed, name, freq, periods):
102102

103103
# defined from start/end/freq
104104
result = interval_range(
105-
start=start, end=end, freq=freq, name=name, closed=closed
105+
start=start, end=end, freq=freq, name=name, inclusive=closed
106106
)
107107
tm.assert_index_equal(result, expected)
108108

109109
# defined from start/periods/freq
110110
result = interval_range(
111-
start=start, periods=periods, freq=freq, name=name, closed=closed
111+
start=start, periods=periods, freq=freq, name=name, inclusive=closed
112112
)
113113
tm.assert_index_equal(result, expected)
114114

115115
# defined from end/periods/freq
116116
result = interval_range(
117-
end=end, periods=periods, freq=freq, name=name, closed=closed
117+
end=end, periods=periods, freq=freq, name=name, inclusive=closed
118118
)
119119
tm.assert_index_equal(result, expected)
120120

121121
# GH 20976: linspace behavior defined from start/end/periods
122122
result = interval_range(
123-
start=start, end=end, periods=periods, name=name, closed=closed
123+
start=start, end=end, periods=periods, name=name, inclusive=closed
124124
)
125125
tm.assert_index_equal(result, expected)
126126

pandas/tests/indexes/interval/test_setops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ def test_set_incompatible_types(self, closed, op_name, sort):
194194
tm.assert_index_equal(result, expected)
195195

196196
# GH 19016: incompatible dtypes -> cast to object
197-
other = interval_range(Timestamp("20180101"), periods=9, closed=closed)
197+
other = interval_range(Timestamp("20180101"), periods=9, inclusive=closed)
198198
expected = getattr(index.astype(object), op_name)(other, sort=sort)
199199
if op_name == "difference":
200200
expected = index

pandas/tests/indexing/test_coercion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -709,7 +709,7 @@ def test_fillna_datetime64tz(self, index_or_series, fill_val, fill_dtype):
709709
],
710710
)
711711
def test_fillna_interval(self, index_or_series, fill_val):
712-
ii = pd.interval_range(1.0, 5.0, closed="right").insert(1, np.nan)
712+
ii = pd.interval_range(1.0, 5.0, inclusive="right").insert(1, np.nan)
713713
assert isinstance(ii.dtype, pd.IntervalDtype)
714714
obj = index_or_series(ii)
715715

pandas/tests/indexing/test_loc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1508,12 +1508,12 @@ def test_loc_getitem_interval_index(self):
15081508

15091509
def test_loc_getitem_interval_index2(self):
15101510
# GH#19977
1511-
index = pd.interval_range(start=0, periods=3, closed="both")
1511+
index = pd.interval_range(start=0, periods=3, inclusive="both")
15121512
df = DataFrame(
15131513
[[1, 2, 3], [4, 5, 6], [7, 8, 9]], index=index, columns=["A", "B", "C"]
15141514
)
15151515

1516-
index_exp = pd.interval_range(start=0, periods=2, freq=1, closed="both")
1516+
index_exp = pd.interval_range(start=0, periods=2, freq=1, inclusive="both")
15171517
expected = Series([1, 4], index=index_exp, name="A")
15181518
result = df.loc[1, "A"]
15191519
tm.assert_series_equal(result, expected)

pandas/tests/util/test_assert_interval_array_equal.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
[
1010
{"start": 0, "periods": 4},
1111
{"start": 1, "periods": 5},
12-
{"start": 5, "end": 10, "closed": "left"},
12+
{"start": 5, "end": 10, "inclusive": "left"},
1313
],
1414
)
1515
def test_interval_array_equal(kwargs):
@@ -19,8 +19,9 @@ def test_interval_array_equal(kwargs):
1919

2020
def test_interval_array_equal_closed_mismatch():
2121
kwargs = {"start": 0, "periods": 5}
22-
arr1 = interval_range(closed="left", **kwargs).values
23-
arr2 = interval_range(closed="right", **kwargs).values
22+
arr1 = interval_range(inclusive="left", **kwargs).values
23+
arr1 = interval_range(inclusive="left", **kwargs).values
24+
arr2 = interval_range(inclusive="right", **kwargs).values
2425

2526
msg = """\
2627
IntervalArray are different

0 commit comments

Comments
 (0)