Skip to content

Commit 22ab15b

Browse files
chean.wei.khorchean.wei.khor
authored andcommitted
test
1 parent 1c41aa0 commit 22ab15b

File tree

6 files changed

+12
-10
lines changed

6 files changed

+12
-10
lines changed

pandas/tests/arrays/categorical/test_constructors.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -689,7 +689,7 @@ def test_constructor_string_and_tuples(self):
689689
assert c.categories.equals(expected_index)
690690

691691
def test_interval(self):
692-
idx = pd.interval_range(0, 10, periods=10)
692+
idx = pd.interval_range(0, 10, periods=10, inclusive="right")
693693
cat = Categorical(idx, categories=idx)
694694
expected_codes = np.arange(10, dtype="int8")
695695
tm.assert_numpy_array_equal(cat.codes, expected_codes)
@@ -717,7 +717,7 @@ def test_interval(self):
717717
tm.assert_index_equal(cat.categories, idx)
718718

719719
# extra
720-
values = pd.interval_range(8, 11, periods=3)
720+
values = pd.interval_range(8, 11, periods=3, inclusive="right")
721721
cat = Categorical(values, categories=idx)
722722
expected_codes = np.array([8, 9, -1], dtype="int8")
723723
tm.assert_numpy_array_equal(cat.codes, expected_codes)

pandas/tests/frame/methods/test_round.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ def test_round_nonunique_categorical(self):
210210

211211
def test_round_interval_category_columns(self):
212212
# GH#30063
213-
columns = pd.CategoricalIndex(pd.interval_range(0, 2))
213+
columns = pd.CategoricalIndex(pd.interval_range(0, 2, inclusive="right"))
214214
df = DataFrame([[0.66, 1.1], [0.3, 0.25]], columns=columns)
215215

216216
result = df.round()

pandas/tests/indexes/interval/test_astype.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def test_subtype_integer(self, subtype_start, subtype_end):
117117
@pytest.mark.xfail(reason="GH#15832")
118118
def test_subtype_integer_errors(self):
119119
# int64 -> uint64 fails with negative values
120-
index = interval_range(-10, 10)
120+
index = interval_range(-10, 10, inclusive="right")
121121
dtype = IntervalDtype("uint64", "right")
122122

123123
# Until we decide what the exception message _should_ be, we
@@ -170,7 +170,7 @@ def test_subtype_integer_with_non_integer_borders(self, subtype):
170170

171171
def test_subtype_integer_errors(self):
172172
# float64 -> uint64 fails with negative values
173-
index = interval_range(-10.0, 10.0)
173+
index = interval_range(-10.0, 10.0, inclusive="right")
174174
dtype = IntervalDtype("uint64", "right")
175175
msg = re.escape(
176176
"Cannot convert interval[float64, right] to interval[uint64, right]; "

pandas/tests/indexes/interval/test_interval_range.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,9 @@ def test_no_invalid_float_truncation(self, start, end, freq):
163163
breaks = [0.5, 2.0, 3.5, 5.0, 6.5]
164164
expected = IntervalIndex.from_breaks(breaks)
165165

166-
result = interval_range(start=start, end=end, periods=4, freq=freq)
166+
result = interval_range(
167+
start=start, end=end, periods=4, freq=freq, inclusive="right"
168+
)
167169
tm.assert_index_equal(result, expected)
168170

169171
@pytest.mark.parametrize(
@@ -184,7 +186,7 @@ def test_no_invalid_float_truncation(self, start, end, freq):
184186
def test_linspace_dst_transition(self, start, mid, end):
185187
# GH 20976: linspace behavior defined from start/end/periods
186188
# accounts for the hour gained/lost during DST transition
187-
result = interval_range(start=start, end=end, periods=2)
189+
result = interval_range(start=start, end=end, periods=2, inclusive="right")
188190
expected = IntervalIndex.from_breaks([start, mid, end])
189191
tm.assert_index_equal(result, expected)
190192

pandas/tests/indexing/test_loc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1497,7 +1497,7 @@ def test_loc_setitem_2d_to_1d_raises(self):
14971497

14981498
def test_loc_getitem_interval_index(self):
14991499
# GH#19977
1500-
index = pd.interval_range(start=0, periods=3)
1500+
index = pd.interval_range(start=0, periods=3, inclusive="right")
15011501
df = DataFrame(
15021502
[[1, 2, 3], [4, 5, 6], [7, 8, 9]], index=index, columns=["A", "B", "C"]
15031503
)

pandas/tests/series/indexing/test_setitem.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -779,7 +779,7 @@ def test_index_putmask(self, obj, key, expected, val):
779779
pytest.param(
780780
# GH#45568 setting a valid NA value into IntervalDtype[int] should
781781
# cast to IntervalDtype[float]
782-
Series(interval_range(1, 5)),
782+
Series(interval_range(1, 5, inclusive="right")),
783783
Series(
784784
[Interval(1, 2), np.nan, Interval(3, 4), Interval(4, 5)],
785785
dtype="interval[float64]",
@@ -1356,7 +1356,7 @@ class TestPeriodIntervalCoercion(CoercionTest):
13561356
@pytest.fixture(
13571357
params=[
13581358
period_range("2016-01-01", periods=3, freq="D"),
1359-
interval_range(1, 5),
1359+
interval_range(1, 5, inclusive="right"),
13601360
]
13611361
)
13621362
def obj(self, request):

0 commit comments

Comments
 (0)