Skip to content

Commit d3efb23

Browse files
committed
Separate the test and add more cases
1 parent efa30c7 commit d3efb23

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pandas/tests/extension/base/reshaping.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -173,14 +173,15 @@ def test_merge(self, data, na_value):
173173
dtype=data.dtype)})
174174
self.assert_frame_equal(res, exp[['ext', 'int1', 'key', 'int2']])
175175

176+
@pytest.mark.parametrize("dtypes", ["Int64"])
177+
def test_merge_on_int_array(self, data, dtypes):
176178
# GH 23020
177179
df1 = pd.DataFrame({'ext': data[:3],
178-
'key': pd.Series([1, 2, np.nan], dtype='Int64')})
179-
180+
'key': pd.Series([1, 2, np.nan], dtype=dtypes)})
180181
res = pd.merge(df1, df1, on='key')
181182

182183
exp = pd.DataFrame(
183-
{'key': pd.Series([1, 2, np.nan], dtype='Int64'),
184+
{'key': pd.Series([1, 2, np.nan], dtype=dtypes),
184185
'ext_x': data._from_sequence(data[:3], dtype=data.dtype),
185186
'ext_y': data._from_sequence(data[:3], dtype=data.dtype)})
186187

0 commit comments

Comments
 (0)