@@ -4350,6 +4350,20 @@ def test_unstack() -> None:
4350
4350
["b" , "b" , pd .Timestamp (2024 , 3 , 2 )],
4351
4351
]
4352
4352
).set_index ([0 , 1 ])
4353
+ df_sr = pd .DataFrame (
4354
+ [
4355
+ ["a" , "b" , "abc" ],
4356
+ ["a" , "a" , "def" ],
4357
+ ["b" , "b" , "ghi" ],
4358
+ ]
4359
+ ).set_index ([0 , 1 ])
4360
+ df_flt = pd .DataFrame (
4361
+ [
4362
+ ["a" , "b" , 1 ],
4363
+ ["a" , "a" , 12 ],
4364
+ ["b" , "b" , 14 ],
4365
+ ]
4366
+ ).set_index ([0 , 1 ])
4353
4367
4354
4368
check (assert_type (df .unstack (0 ), pd .DataFrame | pd .Series ), pd .DataFrame )
4355
4369
check (
@@ -4359,14 +4373,15 @@ def test_unstack() -> None:
4359
4373
pd .DataFrame ,
4360
4374
)
4361
4375
check (
4362
- assert_type (df .unstack (1 , fill_value = 0.0 ), pd .DataFrame | pd .Series ),
4376
+ assert_type (df_flt .unstack (1 , fill_value = 0.0 ), pd .DataFrame | pd .Series ),
4363
4377
pd .DataFrame ,
4364
4378
)
4365
4379
check (
4366
- assert_type (df .unstack (1 , fill_value = 1 ), pd .DataFrame | pd .Series ), pd .DataFrame
4380
+ assert_type (df_flt .unstack (1 , fill_value = 1 ), pd .DataFrame | pd .Series ),
4381
+ pd .DataFrame ,
4367
4382
)
4368
4383
check (
4369
- assert_type (df .unstack (1 , fill_value = "string" ), pd .DataFrame | pd .Series ),
4384
+ assert_type (df_sr .unstack (1 , fill_value = "string" ), pd .DataFrame | pd .Series ),
4370
4385
pd .DataFrame ,
4371
4386
)
4372
4387
check (
@@ -4381,7 +4396,7 @@ def test_unstack() -> None:
4381
4396
)
4382
4397
check (
4383
4398
assert_type (
4384
- df .unstack (1 , fill_value = 0.0 , sort = False ), pd .DataFrame | pd .Series
4399
+ df_flt .unstack (1 , fill_value = 0.0 , sort = False ), pd .DataFrame | pd .Series
4385
4400
),
4386
4401
pd .DataFrame ,
4387
4402
)
0 commit comments