File size: 637 Bytes
a366dd4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | from pandas import Index
import pandas._testing as tm
from pandas.core.construction import extract_array
def test_extract_array_rangeindex():
ri = Index(range(5))
expected = ri._values
res = extract_array(ri, extract_numpy=True, extract_range=True)
tm.assert_numpy_array_equal(res, expected)
res = extract_array(ri, extract_numpy=False, extract_range=True)
tm.assert_numpy_array_equal(res, expected)
res = extract_array(ri, extract_numpy=True, extract_range=False)
tm.assert_index_equal(res, ri)
res = extract_array(ri, extract_numpy=False, extract_range=False)
tm.assert_index_equal(res, ri)
|