Added test to check the dtype of the result of take method. Modified take method to return correct dtype.

Author: aijams

pandas/core/arrays/_mixins.py

@@ -173,7 +173,8 @@ def take(
             fill_value=fill_value,
             axis=axis,
         )
-        return self._from_backing_data(new_data)
+
+        return type(self)._simple_new(new_data, new_data.dtype)
 
     # ------------------------------------------------------------------------
 

pandas/tests/arrays/numpy_/test_numpy.py

@@ -325,6 +325,15 @@ def test_factorize_unsigned():
     tm.assert_extension_array_equal(res_unique, NumpyExtensionArray(exp_unique))
 
 
+@pytest.mark.parametrize("dtype", [np.uint32, np.uint64, np.int32, np.int64])
+def test_take_assigns_correct_dtype(dtype):
+    array = NumpyExtensionArray(np.array([1, 2, 3], dtype=dtype))
+
+    result = array.take([-1], allow_fill=True)
+
+    assert result.dtype == np.float64
+
+
 # ----------------------------------------------------------------------------
 # Output formatting