added support for registering aliases in unit_registry at time of instantiation

Darren Dale · Darren Dale · commit 27c436ddda69 · 2009-01-12T08:37:04.000-05:00
diff --git a/quantities/units/unitquantity.py b/quantities/units/unitquantity.py
@@ -13,7 +13,7 @@ class UnitQuantity(Quantity):
     _primary_order = 99
     _secondary_order = 0
 
-    def __new__(cls, name, reference_quantity=None):
+    def __new__(cls, name, reference_quantity=None, aliases=[]):
         data = numpy.array(1, dtype='d')
         ret = numpy.ndarray.__new__(
             cls,
@@ -24,7 +24,7 @@ def __new__(cls, name, reference_quantity=None):
         ret.flags.writeable = False
         return ret
 
-    def __init__(self, name, reference_quantity=None):
+    def __init__(self, name, reference_quantity=None, aliases=[]):
         self._name = name
         self._dimensionality = ImmutableDimensionality({self:1})
 
@@ -37,6 +37,8 @@ def __init__(self, name, reference_quantity=None):
         self.__class__._secondary_order += 1
 
         unit_registry[name] = self
+        for alias in aliases:
+            unit_registry[alias] = self
 
     @property
     def format_order(self):
