基本和子類的Python單元測試

Question

我目前有幾個單元測試共享一組通用測試。這裏有一個例子：

import unittest 

class BaseTest(unittest.TestCase): 

    def testCommon(self): 
     print 'Calling BaseTest:testCommon' 
     value = 5 
     self.assertEquals(value, 5) 

class SubTest1(BaseTest): 

    def testSub1(self): 
     print 'Calling SubTest1:testSub1' 
     sub = 3 
     self.assertEquals(sub, 3) 


class SubTest2(BaseTest): 

    def testSub2(self): 
     print 'Calling SubTest2:testSub2' 
     sub = 4 
     self.assertEquals(sub, 4) 

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

上面的輸出是：

Calling BaseTest:testCommon 
.Calling BaseTest:testCommon 
.Calling SubTest1:testSub1 
.Calling BaseTest:testCommon 
.Calling SubTest2:testSub2 
. 
---------------------------------------------------------------------- 
Ran 5 tests in 0.000s 

OK 

有沒有辦法改寫的上方，正好是第一testCommon不叫？

編輯： 而是上面跑5次測試，我希望它只能運行4次測試，2從SubTest1另有2從SubTest2。 Python單元測試似乎是自己運行原始的BaseTest，我需要一種機制來防止這種情況的發生。

Answer 1

使用多重繼承，所以你的類與普通測試本身不會從TestCase繼承。

import unittest 

class CommonTests(object): 
    def testCommon(self): 
     print 'Calling BaseTest:testCommon' 
     value = 5 
     self.assertEquals(value, 5) 

class SubTest1(unittest.TestCase, CommonTests): 

    def testSub1(self): 
     print 'Calling SubTest1:testSub1' 
     sub = 3 
     self.assertEquals(sub, 3) 


class SubTest2(unittest.TestCase, CommonTests): 

    def testSub2(self): 
     print 'Calling SubTest2:testSub2' 
     sub = 4 
     self.assertEquals(sub, 4) 

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

Answer 2

更改BaseTest方法名設置：

class BaseTest(unittest.TestCase): 
    def setUp(self): 
     print 'Calling BaseTest:testCommon' 
     value = 5 
     self.assertEquals(value, 5) 


class SubTest1(BaseTest): 
    def testSub1(self): 
     print 'Calling SubTest1:testSub1' 
     sub = 3 
     self.assertEquals(sub, 3) 


class SubTest2(BaseTest): 
    def testSub2(self): 
     print 'Calling SubTest2:testSub2' 
     sub = 4 
     self.assertEquals(sub, 4) 

輸出：

冉2個試驗在0.000s

調用BaseTest：testCommon調用
SubTest1：testSub1調用
BaseTest：testCommon調用
SubTest2：testSub 2

從documentation：

TestCase.setUp（）
方法調用來 製備測試夾具。在調用 測試方法之前立即調用 ;由 引發的任何異常將被視爲 錯誤，而不是測試失敗。 默認實現什麼都不做。

Answer 3

你想達到什麼目的？如果你有共同的測試代碼（斷言，模板測試等），然後將它們放在沒有前綴test的方法中，所以unittest將不會加載它們。

import unittest 

class CommonTests(unittest.TestCase): 
     def common_assertion(self, foo, bar, baz): 
      # whatever common code 
      self.assertEqual(foo(bar), baz) 

class BaseTest(CommonTests): 

    def testCommon(self): 
     print 'Calling BaseTest:testCommon' 
     value = 5 
     self.assertEquals(value, 5) 

class SubTest1(CommonTests): 

    def testSub1(self): 
     print 'Calling SubTest1:testSub1' 
     sub = 3 
     self.assertEquals(sub, 3) 

class SubTest2(CommonTests): 

    def testSub2(self): 
     print 'Calling SubTest2:testSub2' 
     sub = 4 
     self.assertEquals(sub, 4) 

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

Answer 4

馬修的答案是我需要使用的答案，因爲我仍然在2.5。 但是從2.7開始，你可以在你想要跳過的任何測試方法上使用@ unittest.skip（）裝飾器。

http://docs.python.org/library/unittest.html#skipping-tests-and-expected-failures

你需要實現自己的跳躍裝飾檢查基本類型。以前沒有使用過這個功能，但是從我的頭頂，你可以使用BaseTest爲標記型調理跳躍：

def skipBaseTest(obj): 
    if type(obj) is BaseTest: 
     return unittest.skip("BaseTest tests skipped") 
    return lambda func: func 

Answer 5

馬修·馬歇爾的回答是偉大的，但它需要你繼承每個測試用例都有兩個類，這很容易出錯。相反，我用這個（蟒蛇> = 2.7）：

class BaseTest(unittest.TestCase): 

    @classmethod 
    def setUpClass(cls): 
     if cls is BaseTest: 
      raise unittest.SkipTest("Skip BaseTest tests, it's a base class") 
     super(BaseTest, cls).setUpClass() 

Answer 6

您可以用一條命令解決這個問題：

del(BaseTest) 

因此，代碼是這樣的：

import unittest 

class BaseTest(unittest.TestCase): 

    def testCommon(self): 
     print 'Calling BaseTest:testCommon' 
     value = 5 
     self.assertEquals(value, 5) 

class SubTest1(BaseTest): 

    def testSub1(self): 
     print 'Calling SubTest1:testSub1' 
     sub = 3 
     self.assertEquals(sub, 3) 


class SubTest2(BaseTest): 

    def testSub2(self): 
     print 'Calling SubTest2:testSub2' 
     sub = 4 
     self.assertEquals(sub, 4) 

del(BaseTest) 

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

Answer 7

我想過解決這個問題的一種方法是在使用基類時隱藏測試方法。這樣，測試不會被跳過，因此在許多測試報告工具中，測試結果可以是綠色而不是黃色。

與mixin方法相比，IDE的PyCharm不會抱怨單元測試方法從基類中丟失。

如果基類繼承自該類，則需要覆蓋setUpClass和tearDownClass方法。

class BaseTest(unittest.TestCase): 
    @classmethod 
    def setUpClass(cls): 
     cls._test_methods = [] 
     if cls is BaseTest: 
      for name in dir(cls): 
       if name.startswith('test') and callable(getattr(cls, name)): 
        cls._test_methods.append((name, getattr(cls, name))) 
        setattr(cls, name, lambda self: None) 

    @classmethod 
    def tearDownClass(cls): 
     if cls is BaseTest: 
      for name, method in cls._test_methods: 
       setattr(cls, name, method) 
      cls._test_methods = [] 

Answer 8

不要使用多重繼承，它會咬你later。

相反，你可以將你的基類成單獨的模塊或與空白類包裝它：

import unittest 

class BaseTestCases: 

    class BaseTest(unittest.TestCase): 

     def testCommon(self): 
      print 'Calling BaseTest:testCommon' 
      value = 5 
      self.assertEquals(value, 5) 


class SubTest1(BaseTestCases.BaseTest): 

    def testSub1(self): 
     print 'Calling SubTest1:testSub1' 
     sub = 3 
     self.assertEquals(sub, 3) 


class SubTest2(BaseTestCases.BaseTest): 

    def testSub2(self): 
     print 'Calling SubTest2:testSub2' 
     sub = 4 
     self.assertEquals(sub, 4) 

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

輸出：

Calling BaseTest:testCommon 
.Calling SubTest1:testSub1 
.Calling BaseTest:testCommon 
.Calling SubTest2:testSub2 
. 
---------------------------------------------------------------------- 
Ran 4 tests in 0.001s 

OK 

Answer 9

另一種選擇是不執行

unittest.main() 

而不是你可以使用

suite = unittest.TestLoader().loadTestsFromTestCase(TestClass) 
unittest.TextTestRunner(verbosity=2).run(suite) 

所以你只能在類執行測試TestClass

Answer 10

只需重命名testCommon方法到別的東西。 Unittest（通常）會跳過任何沒有「測試」的東西。

快速和簡單的

import unittest 

    class BaseTest(unittest.TestCase): 

    def methodCommon(self): 
     print 'Calling BaseTest:testCommon' 
     value = 5 
     self.assertEquals(value, 5) 

    class SubTest1(BaseTest): 

     def testSub1(self): 
      print 'Calling SubTest1:testSub1' 
      sub = 3 
      self.assertEquals(sub, 3) 


    class SubTest2(BaseTest): 

     def testSub2(self): 
      print 'Calling SubTest2:testSub2' 
      sub = 4 
      self.assertEquals(sub, 4) 

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

Answer 11

我做了大約比@Vladim P.（https://stackoverflow.com/a/25695512/2451329）相同，但略作修改：

import unittest2 


from some_module import func1, func2 


def make_base_class(func): 

    class Base(unittest2.TestCase): 

     def test_common1(self): 
      print("in test_common1") 
      self.assertTrue(func()) 

     def test_common2(self): 
      print("in test_common1") 
      self.assertFalse(func(42)) 

    return Base 



class A(make_base_class(func1)): 
    pass 


class B(make_base_class(func2)): 

    def test_func2_with_no_arg_return_bar(self): 
     self.assertEqual("bar", func2()) 

和我們走吧。