From d1a7db2ecdfcc63a0fd3c661d50ec58e127e1c3b Mon Sep 17 00:00:00 2001 From: phofl Date: Mon, 9 Nov 2020 19:13:04 +0100 Subject: [PATCH 1/3] TST: Add test for KeyError with MultiIndex --- pandas/tests/indexing/multiindex/test_loc.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pandas/tests/indexing/multiindex/test_loc.py b/pandas/tests/indexing/multiindex/test_loc.py index d79af1ea6b804..2e61c90c6ca74 100644 --- a/pandas/tests/indexing/multiindex/test_loc.py +++ b/pandas/tests/indexing/multiindex/test_loc.py @@ -598,3 +598,10 @@ def test_getitem_loc_commutability(multiindex_year_month_day_dataframe_random_da result = ser[2000, 5] expected = df.loc[2000, 5]["A"] tm.assert_series_equal(result, expected) + + +def test_getitem_non_found_tuple(): + # GH: 25236 + df = pd.DataFrame([[1, 2, 3, 4]], columns=['a', 'b', 'c', 'd']).set_index(['a', 'b', 'c']) + with pytest.raises(KeyError, match=r"\(2\.0, 2\.0, 3\.0\)"): + df.loc[(2., 2., 3.)] From dc6bb823765c5088b44105eb98cf4a7330c2ef4c Mon Sep 17 00:00:00 2001 From: phofl Date: Mon, 9 Nov 2020 19:15:14 +0100 Subject: [PATCH 2/3] Run black --- pandas/tests/indexing/multiindex/test_loc.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pandas/tests/indexing/multiindex/test_loc.py b/pandas/tests/indexing/multiindex/test_loc.py index 2e61c90c6ca74..18cfdd026e74d 100644 --- a/pandas/tests/indexing/multiindex/test_loc.py +++ b/pandas/tests/indexing/multiindex/test_loc.py @@ -602,6 +602,8 @@ def test_getitem_loc_commutability(multiindex_year_month_day_dataframe_random_da def test_getitem_non_found_tuple(): # GH: 25236 - df = pd.DataFrame([[1, 2, 3, 4]], columns=['a', 'b', 'c', 'd']).set_index(['a', 'b', 'c']) + df = pd.DataFrame([[1, 2, 3, 4]], columns=["a", "b", "c", "d"]).set_index( + ["a", "b", "c"] + ) with pytest.raises(KeyError, match=r"\(2\.0, 2\.0, 3\.0\)"): - df.loc[(2., 2., 3.)] + df.loc[(2.0, 2.0, 3.0)] From 1aa943b1ff400461fa540f1abb8048a398be0c44 Mon Sep 17 00:00:00 2001 From: phofl Date: Mon, 9 Nov 2020 19:18:15 +0100 Subject: [PATCH 3/3] Fix pattern --- pandas/tests/indexing/multiindex/test_loc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandas/tests/indexing/multiindex/test_loc.py b/pandas/tests/indexing/multiindex/test_loc.py index 18cfdd026e74d..28f4f97a1cb76 100644 --- a/pandas/tests/indexing/multiindex/test_loc.py +++ b/pandas/tests/indexing/multiindex/test_loc.py @@ -602,7 +602,7 @@ def test_getitem_loc_commutability(multiindex_year_month_day_dataframe_random_da def test_getitem_non_found_tuple(): # GH: 25236 - df = pd.DataFrame([[1, 2, 3, 4]], columns=["a", "b", "c", "d"]).set_index( + df = DataFrame([[1, 2, 3, 4]], columns=["a", "b", "c", "d"]).set_index( ["a", "b", "c"] ) with pytest.raises(KeyError, match=r"\(2\.0, 2\.0, 3\.0\)"):