You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
637 B
18 lines
637 B
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)
|
|
|