python测试样例

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/TH_NUM/article/details/82820136

使用python编写测试样例，对代码进行测试。
代码测试是项目构建过程中必须的步骤，在大型项目中，为了敏捷开发能够顺利进行，必要的代码测试是保证敏捷开发的必要步骤。在别人遇到问题的时候，可以通过你的测试样例，对函数进行分析。

在平时自己的程序中也需要锻炼必要的代码测试，对自己写的每一行代码都必须保证代码的鲁棒性。

import inspect
import itertools
import os
import tempfile
import torch.nn.functional as F
import unittest
import numpy as np
import torch

class TorchFunctionTests(unittest.TestCase):

    def test_topk(self):
        input_tensor = torch.randn((8,2),dtype=torch.float)

        print(input_tensor)
    def test_itertools_product(self):
        A = [1,2,3]
        B = [4,5,6]
        c = itertools.product(A,B)
        for k in c:
            print(k)


    def test_tempfile_mkstemp(self):
        _,fname = tempfile.mkstemp('.pt')

        print(fname)
    

输出：

tensor([[ 0.7120,  1.3318],
        [ 0.2087, -2.4989],
        [ 2.1261,  0.0639],
        [-0.8680, -0.9963],
        [ 0.6319,  0.4820],
        [ 1.2168,  0.2537],
        [-0.9537, -0.8061],
        [ 0.6252, -0.0696]])


Ran 1 test in 0.003s

OK