test_get_value.py 679 B

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