Optimize the happy path of some unit computations with identical units.

labrad · Feb 17, 2017 · cac38e7 · cac38e7
1 parent 61fa17d
commit cac38e7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/labrad/units.py b/labrad/units.py
@@ -282,6 +282,8 @@ def _convert_units(self, other):
         (float, complex, array), and unit is their common unit.
         """
         if isinstance(other, WithUnit):
+            if other.unit is self.unit:
+                return self._value, other._value, self.unit
             if self.isCompatible(other):
                 if self.unit.conversionFactorTo(other.unit) > 1:
                     unit = other.unit
@@ -415,6 +417,8 @@ def _base_value(self):
 
     def __getitem__(self, unit):
         """Return value of physical quantity expressed in new units."""
+        if unit is self.unit:
+            return self._value
         if unit == self.unit:
             return self._value
         factor, offset = self.unit.conversionTupleTo(unit)