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

前几天，在一个群里，一个人问了，这样一个问题。说他面试遇到一个面试官，问他，为啥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()