这个单元测试是否适合我想要实现的目标？

Question

我的函数中定义了一个列表 CountryList[]，我想检查它 不是空的。我将其初始化为空，但稍后在函数中将数据放入其中。 这是我输入的单元测试。

 def assertEmpty(self, CountryList):
      self.assertFalse(CountryList)

 def assertNotEmpty(self, CountryList):
      self.assertTrue(CountryList)

这是我程序中的方法。

def onCountry(self, doc_id):
        if(doc_id==None):
            return

        output_list = self.findBySubjectDocId(doc_id)

        country_list=[]
        for x in output_list:
            country_id=x["visitor_country"]
            if(SHOW_FULL_NAMES):
                country_id=pc.country_alpha2_to_country_name(country_id)
            country_list.append(country_id)

        ts=pd.Series(country_list).value_counts().plot(kind='bar',color='purple')
        plt.xticks(rotation='horizontal')
        plt.xlabel('Country')
        plt.ylabel('Number of Viewers')
        plt.title("Viewers based on Country")
        ts.plot()
        plt.show()

        print("Countries of Visitors:")
        x = []
        y = []
        for k,v in Counter(country_list).items():
            x.append(k)
            y.append(v)
            print(k,"-",v)

您是否建议我以其他方式测试此代码？或者上述测试是否可以接受？

Answer 1

这里有几个建议：

1. 要正确使用 Python 的 unittest 包，您需要创建一个扩展 unittest.TestCase 的类。

2. 将您的测试方法命名为test_* 而不是assert*。

3. 测试只能接受一个参数self。

4. 您可以定义一个setUp() 方法，该方法在每个test_* 方法之前运行。使用它来创建所有测试通用的数据。

5. 测试是代码，就像任何其他代码一样，遵循所有相同的规则。

6. 首先用文字描述您要测试的场景。通常这是“当我用参数 P 调用函数 F 时，结果将是 R”的形式。然后编写测试来复制这个特定场景。

Answer 2

由于country_list 是一个不返回的局部变量，因此您的单元测试无法直接检查它；您的测试助手看起来不错，但您似乎没有任何使用它们的实际测试，因为代码的编写方式难以测试。

使测试更容易的一种方法是将函数重构为如下两部分：

def build_country_list(self, doc_id):
    output_list = self.findBySubjectDocId(doc_id)

    country_list=[]
    for x in output_list:
        country_id=x["visitor_country"]
        if(SHOW_FULL_NAMES):
            country_id=pc.country_alpha2_to_country_name(country_id)
        country_list.append(country_id)
    return country_list

def onCountry(self, doc_id):
    if doc_id is None:
        return
    country_list = self.build_country_list(doc_id)
        
    ts=pd.Series(country_list).value_counts().plot(kind='bar',color='purple')
    plt.xticks(rotation='horizontal')
    plt.xlabel('Country')
    plt.ylabel('Number of Viewers')
    plt.title("Viewers based on Country")
    ts.plot()
    plt.show()

    print("Countries of Visitors:")
    x = []
    y = []
    for k,v in Counter(country_list).items():
        x.append(k)
        y.append(v)
        print(k,"-",v)

现在您可以为build_country_list 编写一个单元测试，以验证您构建列表的逻辑是否正确，与同样绘制和打印数据的onCountry 逻辑分开，例如：

def test_build_country_list(self):
    # probably need to have done some setUp that sets up a test_doc_id?
    self.assertNotEmpty(self.test_instance.build_country_list(self.test_doc_id))

如果你想也测试输出逻辑，你需要模拟输出函数并编写一个测试来验证当onCountry被调用时它们被正确的参数调用.