pint -> quantity (#20)

Author: lucascolley

.github/workflows/ci.yml

@@ -37,7 +37,7 @@ jobs:
           cache: true
           environments: ${{ matrix.environment }}
 
-      - name: pint-array tests
+      - name: quantity-array tests
         run: pixi run --environment=${{ matrix.environment }} tests
 
       - name: array-api-tests

README.md

@@ -1,5 +1,7 @@
-# pint-array
+# quantity-array
 
-Pint quantities with array API standard arrays.
+Quantities with array API standard arrays.
 
-hgrecco/pint#2101
+Currently based on [`pint`](https://github.com/hgrecco/pint), with a view to
+consuming a standardised Quantity API - see
+https://github.com/quantity-dev/discussions.

pixi.lock

@@ -5,18 +5,30 @@ build-backend = "hatchling.build"
 [project]
 authors = [
     { name = "Lucas Colley", email = "[email protected]" },
+    # { name = "Open Source Contributors" },  # https://github.com/pypi/warehouse/issues/14813
 ]
-name = "pint-array"
-description = "Pint quantities with array API standard arrays."
+name = "quantity-array"
+description = "Quantities with array API standard arrays."
 readme = "README.md"
+classifiers = [
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+]
 dynamic = ["version"]
 requires-python = ">=3.10"
 
 [project.urls]
-Homepage = "https://github.com/lucascolley/pint-array"
+Homepage = "https://github.com/quantity-dev/quantity-array"
 
 [tool.hatch]
-version.path = "src/pint_array/__init__.py"
+version.path = "src/quantity_array/__init__.py"
 
 [tool.pixi.project]
 channels = ["https://prefix.dev/conda-forge"]
@@ -28,7 +40,7 @@ array-api-compat = ">=1.10.0,<2"
 array-api-strict = ">=2.2,<3"
 
 [tool.pixi.pypi-dependencies]
-pint-array = { path = ".", editable = true }
+quantity-array = { path = ".", editable = true }
 
 [tool.pixi.feature.dev.dependencies]
 ipython = ">=8.31.0,<9"
@@ -91,7 +103,7 @@ xp-tests.cmd = [
     "--disable-deadline",
     "array_api_tests/",
 ]
-xp-tests.env.ARRAY_API_TESTS_MODULE = "pint_array.array_api_strict"
+xp-tests.env.ARRAY_API_TESTS_MODULE = "quantity_array.array_api_strict"
 xp-tests.cwd = "array-api-tests"
 xp-tests.depends-on = ["clone-xp-tests"]
 
@@ -109,7 +121,7 @@ ci-py313 = ["py313", "tests", "xp-tests"]
 
 [tool.ruff]
 target-version = "py310"
-exclude = ["src/pint_array/funcs.py"]
+exclude = ["src/quantity_array/funcs.py"]
 
 [tool.ruff.lint]
 extend-select = [

pyproject.toml

@@ -1,9 +1,4 @@
-"""
-pint_array
-~~~~~~~~~~
-
-Pint interoperability with array API standard arrays.
-"""
+"""Quantities with array API standard arrays."""
 
 import collections
 import contextlib
@@ -20,21 +15,21 @@
 from pint.facets.plain import MagnitudeT, PlainQuantity
 
 __version__ = "0.0.1.dev0"
-__all__ = ["__version__", "pint_namespace"]
+__all__ = ["__version__", "quantity_namespace"]
 
 
 def __getattr__(name):
     try:
         xp = importlib.import_module(name)
-        mod = pint_namespace(xp)
+        mod = quantity_namespace(xp)
         sys.modules[f"marray.{name}"] = mod
         return mod
     except ModuleNotFoundError as e:
         raise AttributeError(str(e)) from None
 
 
-def pint_namespace(xp):
-    mod = types.ModuleType(f"pint({xp.__name__})")
+def quantity_namespace(xp):
+    mod = types.ModuleType(f"quantity({xp.__name__})")
 
     class ArrayQuantity(Generic[MagnitudeT], PlainQuantity[MagnitudeT]):
         def __init__(self, *args, **kwargs):  # noqa: ARG002
@@ -695,11 +690,10 @@ def fun(x, /, *args, func_str=func_str, **kwargs):
 
         setattr(mod, func_str, fun)
 
-
     def _sqrt(x, /, *args, **kwargs):
         x = asarray(x)
         magnitude = xp.asarray(x.magnitude, copy=True)
-        magnitude = getattr(xp, "sqrt")(magnitude, *args, **kwargs)
+        magnitude = xp.sqrt(magnitude, *args, **kwargs)
         return ArrayUnitQuantity(magnitude, x.units**0.5)
 
     mod.sqrt = _sqrt
@@ -785,9 +779,7 @@ def pow(x1, x2, /, *args, **kwargs):
             if x1.unitless:
                 units = "dimensionless"
             elif not xp.all(x2_magnitude == x2_first_elem_magnitude):
-                extra_msg = (
-                    "The first array must be unitless, or the exponent must be a scalar or an array of all the same value."
-                )
+                extra_msg = "The first array must be unitless, or the exponent must be a scalar or an array of all the same value."
                 raise DimensionalityError(
                     x2.units,
                     "dimensionless",
@@ -837,9 +829,9 @@ def matrix_transpose(x):
     def _sort(x, /, **kwargs):
         x = asarray(x)
         magnitude = xp.asarray(x.magnitude, copy=True)
-        xp_func = getattr(xp, "sort")
+        xp_func = xp.sort
         magnitude = xp_func(magnitude, **kwargs)
-        units = x.units 
+        units = x.units
         return ArrayUnitQuantity(magnitude, units)
 
     mod.sort = _sort
@@ -960,7 +952,7 @@ def clip(x, /, min=None, max=None):
     preface = [
         "The following is the documentation for the corresponding "
         f"attribute of `{xp.__name__}`.",
-        "The behavior on pint-wrapped arrays is the same for dimensionless "
+        "The behavior on quantity-wrapped arrays is the same for dimensionless "
         "quantities, and may differ for quantities with units.\n\n",
     ]
     preface = "\n".join(preface)

src/pint_array/__init__.py renamed to src/quantity_array/__init__.py

@@ -7,9 +7,9 @@
 from pint import DimensionalityError, OffsetUnitCalculusError
 from pint.testsuite import helpers
 
-import pint_array
+import quantity_array
 
-pxp = pint_array.pint_namespace(xp)
+pxp = quantity_array.quantity_namespace(xp)
 
 
 class TestArrayMethods:
@@ -451,7 +451,7 @@ def test_meshgrid_numpy_func(self):
     def test_comparisons(self):
         self.assertNDArrayEqual(
             pxp.asarray(self.q) > 2 * self.ureg.m,
-            xp.asarray([[False, False], [True, True]])
+            xp.asarray([[False, False], [True, True]]),
         )
         self.assertNDArrayEqual(
             pxp.asarray(self.q) < 2 * self.ureg.m,

tests/test_array.py

@@ -1,7 +1,7 @@
 import importlib.metadata
 
-import pint_array
+import quantity_array
 
 
 def test_version():
-    assert importlib.metadata.version("pint_array") == pint_array.__version__
+    assert importlib.metadata.version("quantity_array") == quantity_array.__version__