python – 如何测试functools.partial产生预期的函数对象

当从一个API转到另一个API时,有时可以帮助在每个API中的相似关键字之间进行映射,允许一个控制器API灵活地分派到其他库,而无需用户在引擎盖下使用不同的API.

假设一些库,other_api,有一个名为“logarithm”的方法,而base的关键字参数是我需要从我的代码中分解出来的东西,比如“log_base_val”;所以要从other_api中使用它我需要键入(例如):

other_api.logarithm(log_base_val=math.e)

考虑像这样的玩具类:

import other_api
import math
import functools

class Foo(object):
    _SUPPORTED_ARGS = {"base":"log_base_val"}

    def arg_binder(self, other_api_function_name, **kwargs):
        other_api_function = getattr(other_api, other_api_function_name)
        other_api_kwargs = {_SUPPORTED_ARGS[k]:v for k,v in kwargs.iteritems()}
        return functools.partial(other_api_function, **other_api_kwargs)

使用Foo,我可以映射一些其他API,其中此参数始终称为base,如下所示:

f = Foo()
ln = f.arg_binder("logarithm", base=math.e)

和ln在逻辑上等价(在kwargs中,来自functools的log_base_val = math.e):

other_api.logarithm(*args, **kwargs)

但是,通过调用functools手动创建相同的参数绑定将导致不同的函数对象:

In [10]: import functools

In [11]: def foo(a, b):
   ....:     return a + b
   ....: 

In [12]: f1 = functools.partial(foo, 2)

In [13]: f2 = functools.partial(foo, 2)

In [14]: id(f1)
Out[14]: 67615304

In [15]: id(f2)
Out[15]: 67615568

因此,对f1 == f2的测试不会按预期成功:

In [16]: f1 == f2
Out[16]: False

所以问题是:测试参数绑定函数是否导致正确的输出函数对象的规定方法是什么?

解决方法:

partial()对象的func属性是对原始函数对象的引用:

f1.func is f2.func

函数对象本身不实现__eq__方法,因此您也可以使用is来测试身份.

类似地,partial().args和partial().关键字包含在调用时传递给函数的参数和关键字参数.

演示:

>>> from functools import partial
>>> def foo(a, b):
...     return a + b
... 
>>> f1 = partial(foo, 2)
>>> f2 = partial(foo, 2)
>>> f1.func is f2.func
True
>>> f1.args
(2,)
>>> f2.args
(2,)
>>> f1.keywords is None
True
>>> f2.keywords is None
True
上一篇:python中的偏函数


下一篇:python--partial偏函数