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],