@@ -256,20 +256,18 @@ def test_fillna_categorical_nan(self):
256256 idx = TimedeltaIndex (["1 days" , "2 days" , "1 days" , NaT , NaT ])
257257 df = DataFrame ({"a" : Categorical (idx )})
258258 tm .assert_frame_equal (df .fillna (value = NaT ), df )
259-
259+
260260 def test_fillna_with_categorical_series (self ):
261- df = DataFrame ({
262- 'cats' : Categorical (['A' , 'B' , 'C' ]),
263- 'ints' : [1.0 , 2.0 , np .nan ]
264- })
265-
261+ df = DataFrame (
262+ {"cats" : Categorical (["A" , "B" , "C" ]), "ints" : [1.0 , 2.0 , np .nan ]}
263+ )
264+
266265 filler = Series (Categorical ([10.0 , 20.0 , 30.0 ]))
267- df .fillna ({'ints' : filler }, inplace = True )
268-
269- expected = DataFrame ({
270- 'cats' : Categorical (['A' , 'B' , 'C' ]),
271- 'ints' : [1.0 , 2.0 , 30.0 ]
272- })
266+ df .fillna ({"ints" : filler }, inplace = True )
267+
268+ expected = DataFrame (
269+ {"cats" : Categorical (["A" , "B" , "C" ]), "ints" : [1.0 , 2.0 , 30.0 ]}
270+ )
273271 tm .assert_frame_equal (df , expected )
274272
275273 def test_fillna_no_downcast (self , frame_or_series ):
0 commit comments