def test_where_putmask_range_cast(self):
# GH#43240
idx = RangeIndex(0, 5, name="test")
mask = np.array([True, True, False, False, False])
result = idx.putmask(mask, 10)
expected = Int64Index([10, 10, 2, 3, 4], name="test")
tm.assert_index_equal(result, expected)
result = idx.where(~mask, 10)
tm.assert_index_equal(result, expected)
def test_where_range_cast(self):
# GH#43240
idx = RangeIndex(0, 5, name="test")
result = idx.where(np.array([False, False, True, True, True]), 10)
expected = Index([10, 10, 2, 3, 4], name="test")
tm.assert_index_equal(result, expected)
