subdtype

hgrecco · Aug 28, 2024 · bc66a5e · bc66a5e
1 parent 99b4c86
commit bc66a5e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/pint_pandas/pint_array.py b/pint_pandas/pint_array.py
@@ -864,15 +864,22 @@ def convert_values(param):
  # a TypeError should be raised
  res = op(lvalues, rvalues)
 
+ subdtype = self.dtype.subdtype
+ if "truediv" in op.__name__ and subdtype.kind == "i":
+ if isinstance(subdtype, _NumpyEADtype):
+ subdtype = "float64"
+ else:
+ subdtype = "Float64"
+
  if op.__name__ == "divmod":
  return (
- cls.from_1darray_quantity(res[0], self.dtype.subdtype),
- cls.from_1darray_quantity(res[1], self.dtype.subdtype),
+ cls.from_1darray_quantity(res[0], subdtype),
+ cls.from_1darray_quantity(res[1], subdtype),
  )
 
  if coerce_to_dtype:
  try:
- res = cls.from_1darray_quantity(res, self.dtype.subdtype)
+ res = cls.from_1darray_quantity(res, subdtype)
  except TypeError:
  pass
 

diff --git a/pint_pandas/testsuite/test_issues.py b/pint_pandas/testsuite/test_issues.py
@@ -41,7 +41,7 @@ def test_force_ndarray_like(self):
 
  result = pd.concat([a, b], axis=1)
  expected = pd.DataFrame(
- {0: PintArray(q_a_), 1: PintArray(q_b)}, dtype="pint[degC][Int64]"
+ {0: PintArray(q_a_), 1: PintArray(q_b)}, dtype="pint[degC][Int32]"
  )
  tm.assert_equal(result, expected)
 
@@ -64,7 +64,7 @@ def test_offset_concat(self):
 
  result = pd.concat([a, b], axis=1)
  expected = pd.DataFrame(
- {0: PintArray(q_a_), 1: PintArray(q_b)}, dtype="pint[degC][Int64]"
+ {0: PintArray(q_a_), 1: PintArray(q_b)}, dtype="pint[degC][Int32]"
  )
  tm.assert_equal(result, expected)