u 3.0__tar.gz → 3.1__tar.gz

{u-3.0 → u-3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: u
-Version: 3.0
+Version: 3.1
 Summary: Statically typed units
 Keywords: units,dimensions
 Author-email: Aran-Fey <rawing7@gmail.com>

{u-3.0 → u-3.1}/changelog.md

@@ -1,9 +1,14 @@
+# 3.1
+
+- Add `__hash__` method
+
 # 3.0
 
 - Add support for `decimal.Decimal`. Unlike floats and Decimals, which throw a TypeError when used
   together, `Quantity`s remain interoperable regardless of whether they were created with a float or
   a Decimal.
 - `Unit.multiplier` is now a Decimal.
+- Add `systems` parameter and attribute to `Unit`
 
 # 2.1
 

{u-3.0 → u-3.1}/tests/test_quantities.py

@@ -8,3 +8,8 @@ def test_getitem():
 
 def test_equality():
     assert u.minutes(60) == u.hours(1)
+
+
+def test_hashing():
+    mapping = {u.minutes(60): "foo"}
+    assert mapping[u.hours(1)] == "foo"

{u-3.0 → u-3.1}/u/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.0"
+__version__ = "3.1"
 
 from .prefixes import *
 from .quantity import *

{u-3.0 → u-3.1}/u/quantity.py

@@ -417,6 +417,14 @@ class Quantity(t.Generic[Q_co], metaclass=QuantityMeta):
     def __neg__(self) -> Quantity[Q_co]:
         return Quantity(-self._value, self._unit)
 
+    # TODO: Rethink equality. I threw in `math.isclose` because I expected problems with floating
+    # point precision, but failed to realize that this breaks the "transitive" property of equality.
+    # (It's now possible to have 3 quantities A, B, C where B is equal to A and C, but A is not
+    # equal to C.)
+    def __hash__(self) -> int:
+        rounded_value = round(self._value, 10)
+        return hash((rounded_value, self._unit))
+
     def __eq__(self, quantity: object, /) -> bool:
         if quantity == 0:
             num = float(self)