openstack
/
deb-python-pint
Code Issues Proposed changes

Merge pull request #458 from eric-wieser/rename-huh 

Rename `huh` to `out_i`, meaning "output index"
This commit is contained in:
Hernan Grecco 2016-11-18 01:55:55 -03:00 committed by GitHub
parent c3291eea39 e85bc50b7f
commit 3a98ac2d4e
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pint/quantity.py
View File

@ -1229,9 +1229,9 @@ class _Quantity(SharedRegistryObject):
# In ufuncs with multiple outputs, domain indicates which output # In ufuncs with multiple outputs, domain indicates which output
# is currently being prepared (eg. see modf). # is currently being prepared (eg. see modf).
# In ufuncs with a single output, domain is 0 # In ufuncs with a single output, domain is 0
uf, objs, huh = context uf, objs, i_out = context
if uf.__name__ in self.__handled and huh == 0: if uf.__name__ in self.__handled and i_out == 0:
# Only one ufunc should be handled at a time. # Only one ufunc should be handled at a time.
# If a ufunc is already being handled (and this is not another domain), # If a ufunc is already being handled (and this is not another domain),
# something is wrong.. # something is wrong..
@ -1244,18 +1244,18 @@ class _Quantity(SharedRegistryObject):
return obj return obj
def __array_wrap__(self, obj, context=None): def __array_wrap__(self, obj, context=None):
uf, objs, huh = context uf, objs, i_out = context
# if this ufunc is not handled by Pint, pass it to the magnitude. # if this ufunc is not handled by Pint, pass it to the magnitude.
if uf.__name__ not in self.__handled: if uf.__name__ not in self.__handled:
return self.magnitude.__array_wrap__(obj, context) return self.magnitude.__array_wrap__(obj, context)
try: try:
ufname = uf.__name__ if huh == 0 else '{0}__{1}'.format(uf.__name__, huh) ufname = uf.__name__ if i_out == 0 else '{0}__{1}'.format(uf.__name__, i_out)
# First, we check the units of the input arguments. # First, we check the units of the input arguments.
if huh == 0: if i_out == 0:
# Do this only when the wrap is called for the first ouput. # Do this only when the wrap is called for the first ouput.
# Store the destination units # Store the destination units
@ -1306,7 +1306,7 @@ class _Quantity(SharedRegistryObject):
out = uf(*mobjs) out = uf(*mobjs)
# If there are multiple outputs, # If there are multiple outputs,
# store them in __handling (uf, objs, huh, out0, out1, ...) # store them in __handling (uf, objs, i_out, out0, out1, ...)
# and return the first # and return the first
if uf.nout > 1: if uf.nout > 1:
self.__handling += out self.__handling += out
@ -1314,7 +1314,7 @@ class _Quantity(SharedRegistryObject):
else: else:
# If this is not the first output, # If this is not the first output,
# just grab the result that was previously calculated. # just grab the result that was previously calculated.
out = self.__handling[3 + huh] out = self.__handling[3 + i_out]
# Second, we set the units of the output value. # Second, we set the units of the output value.
if ufname in self.__set_units: if ufname in self.__set_units:
@ -1352,7 +1352,7 @@ class _Quantity(SharedRegistryObject):
finally: finally:
# If this is the last output argument for the ufunc, # If this is the last output argument for the ufunc,
# we are done handling this ufunc. # we are done handling this ufunc.
if uf.nout == huh + 1: if uf.nout == i_out + 1:
self.__handling = None self.__handling = None
return self.magnitude.__array_wrap__(obj, context) return self.magnitude.__array_wrap__(obj, context)