Как автоматически создать номер тестового набора в unittest? - PullRequest
Новая
       8

Как автоматически создать номер тестового набора в unittest?

0 голосов
/ 05 октября 2019

как я могу автоматически сгенерировать количество тестов в unittest? Я имею в виду что-то вроде test_01, test_02, test_ {генерируем число}. 

import unittest 

class TestSum(unittest.TestCase): 
    def test_01_sum(self): 
        self.assertEqual(sum([1, 2, 3]), 6, "Should be 6") 

    def test_02_sum_tuple(self): 
        self.assertEqual(sum((1, 2, 2)), 6, "Should be 6") 

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

1 Ответ

2 голосов
/ 05 октября 2019

Чтобы достичь этого во время выполнения, вы можете переименовать методы тестирования для своего класса тестирования: 

def generate_test_numbers(test_class):
    counter = 1
    for method_name in dir(test_class):
        if not method_name.startswith('test_N_'):
            continue
        method = getattr(test_class, method_name)
        if not callable(method):
            continue
        new_method_name = method_name.replace('_N_', '_{:02d}_'.format(counter))
        counter += 1
        setattr(test_class, new_method_name, method)
        delattr(test_class, method_name)
    return test_class

Вы можете просто вызвать эту функцию из main: 

generate_test_numbers(TestSum)

иликак предложил @VPfB, используйте его в качестве декоратора класса: 

@generate_test_numbers
class TestSum(unittest.TestCase): 
    def test_N_sum(self): 
        self.assertEqual(sum([1, 2, 3]), 6, "Should be 6") 

    def test_N_sum_tuple(self): 
        self.assertEqual(sum((1, 2, 2)), 6, "Should be 6")

Любой из них выведет (с -v) 

test_01_sum (__main__.TestSum) ... ok
test_02_sum_tuple (__main__.TestSum) ... FAIL

======================================================================
FAIL: test_02_sum_tuple (__main__.TestSum)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "q.py", line 8, in test_N_sum_tuple
    self.assertEqual(sum((1, 2, 2)), 6, "Should be 6")
AssertionError: 5 != 6 : Should be 6

----------------------------------------------------------------------
Ran 2 tests in 0.001s

FAILED (failures=1)
...