diff --git a/pandas/tests/frame/test_indexing.py b/pandas/tests/frame/test_indexing.py index 3b8daa28227f8..4c1abfb1a7f6f 100644 --- a/pandas/tests/frame/test_indexing.py +++ b/pandas/tests/frame/test_indexing.py @@ -76,8 +76,9 @@ def test_get_none(self, df): # see gh-5652 assert df.get(None) is None - def test_loc_iterable(self, float_frame): - idx = iter(['A', 'B', 'C']) + @pytest.mark.parametrize('key_type', [iter, np.array, Series, Index]) + def test_loc_iterable(self, float_frame, key_type): + idx = key_type(['A', 'B', 'C']) result = float_frame.loc[:, idx] expected = float_frame.loc[:, ['A', 'B', 'C']] assert_frame_equal(result, expected)