add more types into the test_merge_on_int_array

Author: makbigc

pandas/core/reshape/merge.py

@@ -1616,8 +1616,8 @@ def _factorize_keys(lk, rk, sort=True):
  is_extension_array_dtype(rk) and
  lk.dtype == rk.dtype):
  klass = libhashtable.Factorizer
- lk = ensure_object(lk)
- rk = ensure_object(rk)
+ lk, _ = lk._values_for_factorize()
+ rk, _ = rk._values_for_factorize()
  elif is_integer_dtype(lk) and is_integer_dtype(rk):
  # GH#23917 TODO: needs tests for case where lk is integer-dtype
  # and rk is datetime-dtype
@@ -1627,8 +1627,9 @@ def _factorize_keys(lk, rk, sort=True):
  elif (issubclass(lk.dtype.type, (np.timedelta64, np.datetime64)) and
  issubclass(rk.dtype.type, (np.timedelta64, np.datetime64))):
  # GH#23917 TODO: Needs tests for non-matching dtypes
- lk, _ = lk._values_for_factorize()
- rk, _ = rk._values_for_factorize()
+ klass = libhashtable.Int64Factorizer
+ lk = ensure_int64(com.values_from_object(lk))
+ rk = ensure_int64(com.values_from_object(rk))
  else:
  klass = libhashtable.Factorizer
  lk = ensure_object(lk)

pandas/tests/extension/base/reshaping.py

@@ -173,7 +173,8 @@ def test_merge(self, data, na_value):
  dtype=data.dtype)})
  self.assert_frame_equal(res, exp[['ext', 'int1', 'key', 'int2']])
 
- @pytest.mark.parametrize("dtypes", ["Int64"])
+ @pytest.mark.parametrize("dtypes", ["Int8", "Int16", "Int32", "Int64",
+ "UInt8", "UInt16", "UInt32", "UInt64"])
  def test_merge_on_int_array(self, data, dtypes):
  # GH 23020
  df1 = pd.DataFrame({'ext': data[:3],