@@ -40,22 +40,22 @@ def test_merge_antijoin_on_different_columns(self):
40
40
"A" : [3.0 ],
41
41
"B" : ["c" ],
42
42
"C" : [np .nan ],
43
- "D" : np . array ( [np .nan ], dtype = object ) ,
43
+ "D" : [np .nan ],
44
44
},
45
45
index = [2 ],
46
- )
46
+ ). astype ({ "B" : object , "D" : object })
47
47
tm .assert_frame_equal (result , expected )
48
48
49
49
result = merge (left , right , how = "right_anti" , left_on = "B" , right_on = "D" )
50
50
expected = DataFrame (
51
51
{
52
52
"A" : [np .nan ],
53
- "B" : np . array ( [np .nan ], dtype = object ) ,
53
+ "B" : [np .nan ],
54
54
"C" : [2.0 ],
55
55
"D" : ["d" ],
56
56
},
57
57
index = [1 ],
58
- )
58
+ ). astype ({ "B" : object , "D" : object })
59
59
tm .assert_frame_equal (result , expected )
60
60
61
61
def test_merge_antijoin_nonunique_keys (self ):
@@ -72,22 +72,22 @@ def test_merge_antijoin_nonunique_keys(self):
72
72
"A" : [1.0 ],
73
73
"B" : ["a" ],
74
74
"C" : [np .nan ],
75
- "D" : np . array ( [np .nan ], dtype = object ) ,
75
+ "D" : [np .nan ],
76
76
},
77
77
index = [0 ],
78
- )
78
+ ). astype ({ "B" : object , "D" : object })
79
79
tm .assert_frame_equal (result , expected )
80
80
81
81
result = merge (left , right , how = "right_anti" , left_on = "B" , right_on = "D" )
82
82
expected = DataFrame (
83
83
{
84
84
"A" : [np .nan , np .nan ],
85
- "B" : np . array ( [np .nan , np .nan ], dtype = object ) ,
85
+ "B" : [np .nan , np .nan ],
86
86
"C" : [2.0 , 4.0 ],
87
87
"D" : ["d" , "d" ],
88
88
},
89
89
index = [2 , 3 ],
90
- )
90
+ ). astype ({ "B" : object , "D" : object })
91
91
tm .assert_frame_equal (result , expected )
92
92
93
93
def test_merge_antijoin_same_df (self ):
@@ -104,8 +104,8 @@ def test_merge_antijoin_nans(self):
104
104
{"D" : object }
105
105
)
106
106
result = merge (left , right , how = "left_anti" , on = "A" )
107
- expected = DataFrame (
108
- {"A" : [ 1.0 ], " C" : [ "a" ] , "D" : np . array ([ np . nan ], dtype = object ) }
107
+ expected = DataFrame ({ "A" : [ 1.0 ], "C" : [ "a" ], "D" : [ np . nan ]}). astype (
108
+ {"C" : object , "D" : object }
109
109
)
110
110
tm .assert_frame_equal (result , expected )
111
111
0 commit comments