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)