1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
| 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)
|
|
