23 lines
679 B
Python
23 lines
679 B
Python
|
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
|