Выполнить юнит-тест в классе

Question

У меня есть набор тестов для тестирования дыма. У меня весь сценарий хранится в разных классах, но когда я пытаюсь запустить набор тестов, я не могу заставить его работать, если он в классе. Код ниже: (класс для вызова тестов)

from alltests import SmokeTests 

class CallTests(SmokeTests): 

    def integration(self): 

        self.suite() 

if __name__ == '__main__': 
    run = CallTests() 
    run.integration() 

И набор тестов:

class SmokeTests(): 

    def suite(self): #Function stores all the modules to be tested  
        modules_to_test = ('external_sanity', 'internal_sanity') # This is the name of the file
        alltests = unittest.TestSuite() 
        for module in map(__import__, modules_to_test): 
            alltests.addTest(unittest.findTestCases(module)) 
        return alltests 
if __name__ == '__main__': 
    unittest.main(defaultTest='suite') 

Итак, я вижу, как вызывать определенную обычную функцию, но мне трудно вызывать ее в комплекте. В одном из тестов пакет настроен так:

class TestInternalSanity(unittest.TestCase):

    def setUp(self):

        setUp script ....

    def tearDown(self):

        script .... 

class BasicInternalSanity(TestInternalSanity):

    def runTest(self):

        test script ....

class InternalSanityTestSuite(unittest.TestSuite): 

    # Tests to be tested by test suite 
    def makeInternalSanityTestSuite(): 
        suite = unittest.TestSuite() 
        suite.addTest(TestInternalSanity("BasicInternalSanity")) 
        suite.addTest(TestInternalSanity("VerifyInternalSanityTestFail")) 
        return suite 

    def suite(): 
        return unittest.makeSuite(TestInternalSanity) 

Если у меня def suite() внутри class SmokeTests, скрипт выполняется, но тесты не запускаются, но если я удаляю класс, тесты запускаются. Я запускаю это как скрипт и вызываю переменные в тестах. Я не хочу запускать тесты с помощью os.system ('python tests.py'). Я надеялся вызвать тесты через класс, который у меня есть, как и любую другую функцию. Это нужно вызывать из класса, так как скрипт, из которого я его вызываю, является объектно-ориентированным. Если кто-нибудь может получить код для запуска с помощью тестов вызовов, я был бы очень признателен.

Эта работа:

def suite(): #Function stores all the modules to be tested  
    modules_to_test = ('external_sanity', 'internal_sanity') 
    alltests = unittest.TestSuite() 
    for module in map(__import__, modules_to_test): 
        alltests.addTest(unittest.findTestCases(module)) 
    return alltests 
if __name__ == '__main__': 
    unittest.main(defaultTest='suite') 

Это не работает:

class SmokeTests():

    def suite(self): #Function stores all the modules to be tested  
        modules_to_test = ('external_sanity', 'internal_sanity') 
        alltests = unittest.TestSuite() 
        for module in map(__import__, modules_to_test): 
            alltests.addTest(unittest.findTestCases(module)) 
        return alltests 
if __name__ == '__main__': 
    unittest.main(defaultTest='suite') 

Кажется, я не могу заставить это работать в классе, кто-нибудь может увидеть решение.

Спасибо

Answer 1

Получилось, извините, что тратит время впустую, ответом было изменение имени теста по умолчанию.

class SmokeTests(): 

    def suite(self): #Function stores all the modules to be tested   
        modules_to_test = ('external_sanity', 'internal_sanity')  
        alltests = unittest.TestSuite()  
        for module in map(__import__, modules_to_test):  
            alltests.addTest(unittest.findTestCases(module))  
        return alltests  
if __name__ == '__main__':
    Smoke = SmokeTests()  
    unittest.main(defaultTest='Smoke.suite') 

Спасибо за любую помощь.

Answer 2

Похоже, вы делаете юнит-тесты намного сложнее, чем они есть на самом деле. Возможно, ваша реализация должна выглядеть примерно так:

import unittest

class MyClass(object):

    def add(self, val, other):
        return val + other

    def subtract(self, val, other):
        return val - other


class TestClass(unittest.TestCase):

    def test_add(self):
        myclass = MyClass()
        self.assert_(myclass.add(1, 2) == 3)

    def test_subtract(self):
        myclass = MyClass()
        self.assert_(myclass.subtract(2, 1) == 1)


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