| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485 |
- """
- Tests for ndarray-like method on the base Index class
- """
- import numpy as np
- import pytest
- from pandas import Index
- import pandas._testing as tm
- class TestReshape:
- def test_repeat(self):
- repeats = 2
- index = Index([1, 2, 3])
- expected = Index([1, 1, 2, 2, 3, 3])
- result = index.repeat(repeats)
- tm.assert_index_equal(result, expected)
- def test_insert(self):
- # GH 7256
- # validate neg/pos inserts
- result = Index(["b", "c", "d"])
- # test 0th element
- tm.assert_index_equal(Index(["a", "b", "c", "d"]), result.insert(0, "a"))
- # test Nth element that follows Python list behavior
- tm.assert_index_equal(Index(["b", "c", "e", "d"]), result.insert(-1, "e"))
- # test loc +/- neq (0, -1)
- tm.assert_index_equal(result.insert(1, "z"), result.insert(-2, "z"))
- # test empty
- null_index = Index([])
- tm.assert_index_equal(Index(["a"]), null_index.insert(0, "a"))
- def test_insert_missing(self, nulls_fixture):
- # GH#22295
- # test there is no mangling of NA values
- expected = Index(["a", nulls_fixture, "b", "c"])
- result = Index(list("abc")).insert(1, nulls_fixture)
- tm.assert_index_equal(result, expected)
- @pytest.mark.parametrize(
- "val", [(1, 2), np.datetime64("2019-12-31"), np.timedelta64(1, "D")]
- )
- @pytest.mark.parametrize("loc", [-1, 2])
- def test_insert_datetime_into_object(self, loc, val):
- # GH#44509
- idx = Index(["1", "2", "3"])
- result = idx.insert(loc, val)
- expected = Index(["1", "2", val, "3"])
- tm.assert_index_equal(result, expected)
- assert type(expected[2]) is type(val)
- @pytest.mark.parametrize(
- "pos,expected",
- [
- (0, Index(["b", "c", "d"], name="index")),
- (-1, Index(["a", "b", "c"], name="index")),
- ],
- )
- def test_delete(self, pos, expected):
- index = Index(["a", "b", "c", "d"], name="index")
- result = index.delete(pos)
- tm.assert_index_equal(result, expected)
- assert result.name == expected.name
- def test_delete_raises(self):
- index = Index(["a", "b", "c", "d"], name="index")
- msg = "index 5 is out of bounds for axis 0 with size 4"
- with pytest.raises(IndexError, match=msg):
- index.delete(5)
- def test_append_multiple(self):
- index = Index(["a", "b", "c", "d", "e", "f"])
- foos = [index[:2], index[2:4], index[4:]]
- result = foos[0].append(foos[1:])
- tm.assert_index_equal(result, index)
- # empty
- result = index.append([])
- tm.assert_index_equal(result, index)
|
