38
38
from pandas .api .types import is_scalar
39
39
from pandas .core .api import Float64Index
40
40
from pandas .tests .indexing .common import Base
41
+ import unittest
41
42
42
43
43
44
class TestLoc (Base ):
@@ -2833,7 +2834,7 @@ def test_loc_assign_dict_to_row(self, dtype):
2833
2834
2834
2835
@td .skip_array_manager_invalid_test
2835
2836
def test_loc_setitem_dict_timedelta_multiple_set (self ):
2836
- # GH 16309
2837
+ # GH 16309s
2837
2838
result = DataFrame (columns = ["time" , "value" ])
2838
2839
result .loc [1 ] = {"time" : Timedelta (6 , unit = "s" ), "value" : "foo" }
2839
2840
result .loc [1 ] = {"time" : Timedelta (6 , unit = "s" ), "value" : "foo" }
@@ -2857,3 +2858,15 @@ def test_loc_set_multiple_items_in_multiple_new_columns(self):
2857
2858
)
2858
2859
2859
2860
tm .assert_frame_equal (df , expected )
2861
+
2862
+
2863
+ class TestLoc3 (unittest .TestCase ): #test case for issue #43599
2864
+ def test_loc_multiindex (self ):
2865
+ df = pd .DataFrame (index = pd .MultiIndex .from_product ([list ('abc' ), list ('de' ), list ('f' )]), columns = ['Val' ])
2866
+ df1 = df .loc [np .s_ [:,'d' ,:]]
2867
+ df2 = pd .DataFrame (index = pd .MultiIndex .from_product ([list ('abc' ), list ('d' ), list ('f' )]), columns = ['Val' ])
2868
+
2869
+
2870
+ tm .assert_frame_equal (df1 ,df2 )
2871
+
2872
+
0 commit comments