kind

andrewgsavage · Aug 30, 2024 · a414c31 · a414c31
1 parent e30f1f5
commit a414c31
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/pint_pandas/pint_array.py b/pint_pandas/pint_array.py
@@ -42,7 +42,7 @@ class PintType(ExtensionDtype):
  """
 
  type = _Quantity
- kind = 'O'
+ kind = "O"
  # str = '|O08'
  # base = np.dtype('O')
  # num = 102

diff --git a/pint_pandas/testsuite/test_issues.py b/pint_pandas/testsuite/test_issues.py
@@ -311,3 +311,14 @@ def test_issue246(self):
 
  # now an operation where each cell is independent from each other
  df.apply(lambda x: x * 2, axis=1)
+
+
+def TestIssue245(BaseExtensionTests):
+ def test_issue245(self):
+ km = pd.Series([1.0, 2.0, np.nan], dtype="pint[km]")
+ kg = pd.Series([1.0, 2.0, np.nan], dtype="pint[kg]")
+
+ xx = pd.DataFrame({"a": km, "b": km})
+ yy = pd.DataFrame({"a": kg, "b": kg})
+
+ pd.concat([xx, yy], axis=0)