ansys-units 0.4.dev0__tar.gz → 0.5.dev0__tar.gz

{ansys_units-0.4.dev0 → ansys_units-0.5.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ansys-units
-Version: 0.4.dev0
+Version: 0.5.dev0
 Summary: Pythonic interface for units, unit systems, and unit conversions.
 Author-email: "ANSYS, Inc." <pyansys.core@ansys.com>
 Maintainer-email: "ANSYS, Inc." <pyansys.core@ansys.com>
@@ -17,15 +17,16 @@ Classifier: Programming Language :: Python :: 3.11
 License-File: LICENSE
 Requires-Dist: importlib-metadata>=4.0
 Requires-Dist: pyyaml>=6.0
-Requires-Dist: ansys-sphinx-theme==1.2.6 ; extra == "doc"
-Requires-Dist: quarto-cli==1.6.40 ; extra == "doc"
-Requires-Dist: matplotlib==3.10.0 ; extra == "doc"
+Requires-Dist: ansys-sphinx-theme==1.3.2 ; extra == "doc"
+Requires-Dist: pdf2image==1.17.0 ; extra == "doc"
+Requires-Dist: quarto-cli==1.6.42 ; extra == "doc"
+Requires-Dist: matplotlib==3.10.1 ; extra == "doc"
 Requires-Dist: numpydoc==1.8.0 ; extra == "doc"
 Requires-Dist: Sphinx==7.4.7 ; extra == "doc"
 Requires-Dist: sphinx-copybutton==0.5.2 ; extra == "doc"
-Requires-Dist: sphinx-gallery==0.18.0 ; extra == "doc"
-Requires-Dist: sphinx-toggleprompt==0.5.2 ; extra == "doc"
-Requires-Dist: numpy==2.2.2 ; extra == "lists"
+Requires-Dist: sphinx-gallery==0.19.0 ; extra == "doc"
+Requires-Dist: sphinx-toggleprompt==0.6.0 ; extra == "doc"
+Requires-Dist: numpy==2.2.3 ; extra == "lists"
 Requires-Dist: pytest==8.3.4 ; extra == "tests"
 Requires-Dist: pytest-cov==6.0.0 ; extra == "tests"
 Requires-Dist: pytest-mock==3.14.0 ; extra == "tests"

{ansys_units-0.4.dev0 → ansys_units-0.5.dev0}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "flit_core.buildapi"
 [project]
 # Check https://flit.readthedocs.io/en/latest/pyproject_toml.html for all available sections
 name = "ansys-units"
-version = "0.4.dev0"
+version = "0.5.dev0"
 description = "Pythonic interface for units, unit systems, and unit conversions."
 readme = "README.rst"
 requires-python = ">=3.9,<4"
@@ -30,14 +30,15 @@ dependencies = [
 
 [project.optional-dependencies]
 doc = [
-    "ansys-sphinx-theme==1.2.6",
-    "quarto-cli==1.6.40",
-    "matplotlib==3.10.0",
+    "ansys-sphinx-theme==1.3.2",
+    "pdf2image==1.17.0",
+    "quarto-cli==1.6.42",
+    "matplotlib==3.10.1",
     "numpydoc==1.8.0",
     "Sphinx==7.4.7",
     "sphinx-copybutton==0.5.2",
-    "sphinx-gallery==0.18.0",
-    "sphinx-toggleprompt==0.5.2",
+    "sphinx-gallery==0.19.0",
+    "sphinx-toggleprompt==0.6.0",
 ]
 tests = [
     "pytest==8.3.4",
@@ -46,7 +47,7 @@ tests = [
     "pint==0.24.4",
 ]
 lists = [
-    "numpy==2.2.2",
+    "numpy==2.2.3",
 ]
 
 [project.urls]

{ansys_units-0.4.dev0 → ansys_units-0.5.dev0}/src/ansys/units/quantity.py

@@ -23,6 +23,7 @@
 
 from __future__ import annotations
 
+from collections.abc import Iterable
 import operator
 from typing import Union
 
@@ -37,6 +38,15 @@ try:
 except ImportError:
     _array = None
 
+try:
+    from matplotlib.units import AxisInfo, ConversionInterface, registry
+
+    _ci = ConversionInterface
+    _ai = AxisInfo
+    _registry = registry
+except ModuleNotFoundError:
+    _ci, _ai, _registry = object, None, dict()
+
 
 class Quantity:
     """
@@ -89,6 +99,9 @@ class Quantity:
         ):
             raise ExcessiveParameters()
 
+        self._value = value
+        self.unit = units
+
         if copy_from:
             if value:
                 units = copy_from.units
@@ -310,22 +323,10 @@ class Quantity:
             dims(dimensions={base_dims.SOLID_ANGLE: 1.0}),
         ]:
             return get_si_value(self)
-        raise InvalidFloatUsage()
-
-    def __array__(self):
-        if _array:
-            if isinstance(self.value, (float)):
-                return _array.array([self.value])
-            return self.value
-        else:
-            raise NumPyRequired()
 
     def __getitem__(self, idx):
-        if _array:
-            value = float(self.__array__()[idx])
-            return Quantity(value, self.units)
-        else:
-            raise NumPyRequired()
+        if isinstance(self.value, Iterable):
+            return Quantity(value=float(self.value[idx]), units=self.units)
 
     def __str__(self):
         return f'({self.value}, "{self._unit.name}")'
@@ -525,3 +526,29 @@ class RequiresUniqueDimensions(ValueError):
         super().__init__(
             f"For '{unit.name}' to be added '{other_unit.name}' must be removed."
         )
+
+
+class QuantityConverter(_ci):
+
+    @staticmethod
+    def convert(value, unit, axis):
+        if isinstance(value, Quantity):
+            return value._value
+        else:
+            return [quantity._value for quantity in value]
+
+    @staticmethod
+    def axisinfo(unit, axis):
+        return _ai(label=unit)
+
+    @staticmethod
+    def default_units(x, axis):
+        "Return the default unit for x or None"
+        if isinstance(x, Quantity):
+            attr = getattr(x, "unit", None)
+        else:
+            attr = getattr(x[0], "unit", None)
+        return attr.name if isinstance(attr, Unit) else attr
+
+
+_registry[Quantity] = QuantityConverter()