Unittest 采用 1 个位置参数，但给出了 2 个

Question

我正在尝试对我的代码进行单元测试，但我总是收到以下错误：

错误：test_FlipVertically1D_empty_array（main.TestMarginSampling）

回溯（最后一次调用）：文件“C:\Users\s\anaconda3\lib\unittest\mock.py”，第 1337 行，已修补 return func(*newargs, **newkeywargs) 类型错误：test_FlipVertically1D_empty_array() 接受 1 个位置参数，但给出了 2 个

这是我的 class:

的代码

class FlipVertically1D:   
    @staticmethod 
    def calc(data: np.ndarray) -> np.ndarray:
        return - data

这是我的单元测试代码：

import pandas as pd
import numpy as np
import unittest
from unittest.mock import patch, MagicMock

from ddt import ddt, data, file_data, idata, unpack


from augmentation.augment import FlipVertically1D as FlipV1D
from augmentation.augment import FlipHorizontally1D as FlipH1D
from augmentation.augment import RandomFlipping1D as RandF1D


@ddt
class TestMarginSampling1D(unittest.TestCase):
    
    def set_up(self):
        pass

    
    def tear_down(self):
        pass #del self.y_predict
        

    def test_FlipVertically1D_empty_array(self):
        empty = FlipV1D.calc(np.array([]))
        self.assertTrue(len(empty))
        #self.assertEqual(FlipV1D.calc(), 0)

我首先假设我在某个地方通过了 self 太多，但这并没有解决我的问题

Answer 1

问题可能是 ddt 注释。通常，unittest 测试用例看起来与您的代码完全一样，除了 self.

之外没有其他参数

Example of unittest:

import unittest

class TestStringMethods(unittest.TestCase):

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_split(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello', 'world'])
        # check that s.split fails when the separator is not a string
        with self.assertRaises(TypeError):
            s.split(2)

if __name__ == '__main__':
    unittest.main()

但是对于 ddt，这发生了变化，他们期待第二个参数 value:

https://ddt.readthedocs.io/en/latest/example.html

@ddt
class FooTestCase(unittest.TestCase):
    def test_undecorated(self):
        self.assertTrue(larger_than_two(24))

    @data(3, 4, 12, 23)
    def test_larger_than_two(self, value):
        self.assertTrue(larger_than_two(value))

请注意，第一个示例 (test_undecorated) 看起来像您的代码，因此它看起来应该可以工作，但可能只是不工作并且无论如何都会转换方法。

Unittest 采用 1 个位置参数，但给出了 2 个

Unittest takes 1 positional argument but 2 were given

python

unit-testing