Use alias for unit_registry

pydata · Jun 28, 2023 · 9a06b53 · 9a06b53
1 parent 49e8fa7
commit 9a06b53
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/xarray/tests/test_units.py b/xarray/tests/test_units.py
@@ -75,13 +75,15 @@ def assert_equal(q1, q2):
 
 
 class PintInfo(UnitInfo):
-    unit = unit_registry.m
-    compatible_unit = unit_registry.mm
-    incompatible_unit = unit_registry.s
-    dimensionless = unit_registry.dimensionless
+    ureg = unit_registry
+
+    unit = ureg.m
+    compatible_unit = ureg.mm
+    incompatible_unit = ureg.s
+    dimensionless = ureg.dimensionless
 
-    unit_type = unit_registry.Unit
-    quantity_type = unit_registry.Quantity
+    unit_type = ureg.Unit
+    quantity_type = ureg.Quantity
 
     @staticmethod
     def strip_units(quantity):