unittest加载测试用例名称必须以test开头，是否可以定制化

时间：2019-11-14 11:03:57 阅读：258 评论：0 收藏：0 [点我收藏+]

前几天，在一个群里，一个人问了，这样一个问题。说他面试遇到一个面试官，问他，为啥unittest的测试用例要用test 开头，能不能定制化。他不知道为啥。

看到这个题目，我回答当然可以了，可以用leizi开头。可能你不信，那么今天雷子来给你实现下。看我说的是不是真的。那么我来实现下，我不修改前，看看用leizi开头命名的测试用例是否可以执行。

import  unittest
class testone(unittest.TestCase):   
 　　def leizi1(self):        
　　   self.assertTrue(True)    
　　　def leizi2(self):        
　　    self.assertTrue(True)    
　　  def testone(self):        
　　　　　　self.assertTrue(True)
if __name__=="__main__":
    unittest.main()

运行：

技术分享图片

就运行了一个测试用例。leizi开头的测试用例没有执行。我要变，我要让它只执行leizi开头的测试用例。

经过我的修改，我们看下，还是上面的测试用例，我们看下执行结果

技术分享图片

执行了两个测试用例，到底是不是leizi开头的呢，我们来打印下，看看是否执行的是leizi的用例。

import  unittest
class testone(unittest.TestCase):   
 def setUp(self) -> None:pass    

def tearDown(self) -> None:pass    

def leizi1(self):        
　　print("leizitest")        
　　self.assertTrue(True)    
def leizi2(self):        
　　print("leizitest")        
　　self.assertTrue(True)    

def testone(self):        
　　self.assertTrue(True)

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

结果：

技术分享图片

打印的结果与预期完全一致。那么雷子是怎么做到的，我先来给你解析一下。首先我们使用的unittest.main()加载运行的测试用例，我们可以去找这个main方法，在源码中可以发现

技术分享图片

我们可以看到有这个TestProgram这个里面的，有这么一个函数，那么我们对应的去看看这个里面的方法。

技术分享图片

默认加载的是defaultTestLoader，那么我去看看这里面有没有我们想要的。

技术分享图片

这里我们可以看到是一个类，这里有个定义的 test，我们看看。接下来是怎么用的。

技术分享图片

我们可以搜索下，在这个类里面下面有一个获取测试用例名称的，这个地方判断了测试用例名称调用了这个值，我们可以改下试试，testMethodPrefix的值改成leizi试试。还是上面的测试函数。

import  unittestclass testone(unittest.TestCase):   
　　 def leizi1(self):        
　　　　print("leizitest")        
　　　　self.assertTrue(True)    
　　def leizi2(self):        
　　　　print("leizitest")        
　　　　self.assertTrue(True)    
　　def testone(self):       
　　　　 self.assertTrue(True)

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