python - Python有多个参数值

问题描述

我有一个对象，具有用于实现方法提供（）的对象的依赖注入字段：

class DataContainer():

    PROVIDERS = dict()

    def __init__(self):
        self.provider = None

    def provide(self, *args, **kwargs):
        if self.provider:
            return self.provider.provide(self, *args, **kwargs)
        raise KeyError("No provider selected")

我为它编写了一个自定义 Provider 对象，如下所示：

# INTERFACE:

class Provider(ABC):
    @abstractmethod
    def provide(self, *args, **kwargs):
        pass

    @abstractmethod
    def add_product(self, name: str, product: ProviderProduct, *args, **kwargs) -> None:
        pass

    @abstractmethod
    def remove_product(self, name: str, *args, **kwargs) -> None:
        pass

class ProviderProduct(ABC):
    @abstractmethod
    def configure(self, *args, **kwargs) -> ProviderProduct:
        pass


IMPLEMENTATION:

class TestProvider(Provider):
    REGISTERED = dict()

    def provide(self, product, from_dict, *args, **kwargs):
        if product in TestProvider.REGISTERED:
            return TestProvider.REGISTERED[product].configure(from_dict)
        raise KeyError("{} not in REGISTERED".format(product))

class AbstractTestProduct(ProviderProduct, ABC):
    INDEX = [0]
    COLUMNS = list()

    def configure(self, from_dict: Dict) -> ProviderProduct:
        df = pd.DataFrame(index=self.INDEX, columns=self.COLUMNS)
        df.update(from_dict)
        return df

    def add_product(self, name, product):
        if isinstance(product, AbstractTestProduct):
            TestProvider.REGISTERED[name] = product
        else:
            raise ValueError("Given {} class is not of AbstractTestProduct type".format(product.__class__.__name__))

    def remove_product(self, name):
        if name in TestProvider.REGISTERED.keys():
            del TestProvider.REGISTERED[name]
        else:
            raise KeyError("{} not found in registered list".format(name))

现在，当我为它编写一些单元测试时，出现了问题：

class TestProviderTestSuite(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        class TestProviderProduct1(AbstractTestProduct):
            COLUMNS = ['test_col_1_1', 'test_col_1_2', 'test_col_1_3']

        class TestProviderProduct2(AbstractTestProduct):
            COLUMNS = ['test_col_2_1', 'test_col_2_2', 'test_col_2_3', 'test_col_2_4']

        class TestProviderProduct3(AbstractTestProduct):
            COLUMNS = ['test_col_3_1', 'test_col_3_2', 'test_col_3_3', 'test_col_3_4', 'test_col_3_5']
        
        cls.data_container = DataContainer()
        cls.data_container.register_provider("dataframe", TestProvider())
        cls.data_container.change_provider('dataframe')
        cls.data_container.add_provider_product("a", TestProviderProduct1())
        cls.data_container.add_provider_product("b", TestProviderProduct2())
        cls.data_container.add_provider_product("c", TestProviderProduct3())

    def test_should_provide_empty_product_df_a(self):
        # Given
        # -
        
        # When
        product_df_a = self.data_container.provide(product="a")

        # Then
        self.assertEqual(3, len(product_df_a.columns))
        self.assertEqual(1, len(product_df_a.index))
        self.assertTrue(0 in product_df_a.index)
        self.assertTrue(product_df_a.isnull().all(axis='columns').values[0])

对于上面的测试，我收到以下错误：

self = <utils.providers.test_provider.TestProviderTestSuite testMethod=test_should_provide_empty_product_df_a>

    def test_should_provide_empty_product_df_a(self):
        # Given
        # -
    
        # When
>       product_df_a = self.data_container.provide(product="a")

tests\test_provider.py:43: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <DataContainer.DataContainer object at 0x0000022FF859CB48>, args = (), kwargs = {'product': 'a'}

    def provide(self, *args, **kwargs):
        if self.provider:
>           return self.provider.provide(self, *args, **kwargs)
E           TypeError: provide() got multiple values for argument 'product'

我不明白为什么解释器在收到对 DataContainer 方法 .provide(product="a") 的调用时会确定“产品”关键字参数被传递了两次。

有任何想法吗？

对我来说，这发生在 Python 3.7 上。

标签： pythonpython-3.x

在 DataContainer 中，该函数provide接受参数*args，**kwargs这意味着当您在测试中调用它时，product="a"将收集在kwargs.

这意味着当您使用的provide函数时TestProvider，您不会将 product 作为第一个参数，而是作为关键字参数 (in **kwargs)。第二个product来自位置参数，该参数由您作为参数传递的 self 填充（您确定要将 self 作为第一个参数传递吗？）。

他们有很多方法可以解决这个问题，最简单的方法是让所有provide相同的签名

python - Python有多个参数值

问题描述

解决方案

推荐阅读