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()

结果：

只执行了leizi开头的，我们修改成功了。

回到上面这个面试题。我们在面试的时候都会遇到这种情况，其实这个时候我们如果没有做过或者看过源码，可以回答，在我目前的得到的知识中，就是以test开头的，但是要是定制，我要看下源码，是否可以定制。如果你看过源码，那就直接说 我知道在哪个文件下面的 什么值可以直接修改。

面试不难，面试题也不难。

学会解答，能给出方案来。

唯有变化，应对万变题目。

题目唯一，方案千变万化。

合理思路，方案有理有据。

基于需求，给定方案即可。

关注雷子说测试，每天带你分享不一样的东西。

如果你感觉有用，可以收藏，分享给大家。帮助大家一起成长。

有问题也可以和我交流。qq&微信：952943386