1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
| import pytest
|
| from pandas import (
| DataFrame,
| MultiIndex,
| )
|
|
| class TestGetValue:
| def test_get_set_value_no_partial_indexing(self):
| # partial w/ MultiIndex raise exception
| index = MultiIndex.from_tuples([(0, 1), (0, 2), (1, 1), (1, 2)])
| df = DataFrame(index=index, columns=range(4))
| with pytest.raises(KeyError, match=r"^0$"):
| df._get_value(0, 1)
|
| def test_get_value(self, float_frame):
| for idx in float_frame.index:
| for col in float_frame.columns:
| result = float_frame._get_value(idx, col)
| expected = float_frame[col][idx]
| assert result == expected
|
|