@@ -1546,6 +1546,7 @@ def test_dtype_on_categorical_dates(self):
15461546 result_inner = pd .merge (df , df2 , how = 'inner' , on = ['date' ])
15471547 assert_frame_equal (result_inner , expected_inner )
15481548
1549+ @pytest .mark .parametrize ('ordered' , [True , False ])
15491550 @pytest .mark .parametrize ('category_column,categories,expected_categories' ,
15501551 [([False , True , True , False ], [True , False ],
15511552 [True , False ]),
@@ -1554,20 +1555,19 @@ def test_dtype_on_categorical_dates(self):
15541555 ['True' , 'False' ], ['True' , 'False' ])])
15551556 def test_merging_with_bool_or_int_cateorical_column (self , category_column ,
15561557 categories ,
1557- expected_categories ):
1558+ expected_categories ,
1559+ ordered ):
15581560 # GH 17187
15591561 # merging with a boolean/int categorical column
15601562 df1 = pd .DataFrame ({'id' : [1 , 2 , 3 , 4 ],
15611563 'cat' : category_column })
1562- df1 ['cat' ] = df1 ['cat' ].astype ('category' ,
1563- categories = categories , ordered = True )
1564+ df1 ['cat' ] = df1 ['cat' ].astype (CDT (categories , ordered = ordered ))
15641565 df2 = pd .DataFrame ({'id' : [2 , 4 ], 'num' : [1 , 9 ]})
15651566 result = df1 .merge (df2 )
15661567 expected = pd .DataFrame ({'id' : [2 , 4 ], 'cat' : expected_categories ,
15671568 'num' : [1 , 9 ]})
1568- expected ['cat' ] = expected ['cat' ].astype ('category' ,
1569- categories = categories ,
1570- ordered = True )
1569+ expected ['cat' ] = expected ['cat' ].astype (
1570+ CDT (categories , ordered = ordered ))
15711571 assert_frame_equal (expected , result )
15721572
15731573
0 commit comments