本文共 931 字,大约阅读时间需要 3 分钟。
使用python编写测试样例,对代码进行测试。
代码测试是项目构建过程中必须的步骤,在大型项目中,为了敏捷开发能够顺利进行,必要的代码测试是保证敏捷开发的必要步骤。在别人遇到问题的时候,可以通过你的测试样例,对函数进行分析。在平时自己的程序中也需要锻炼必要的代码测试,对自己写的每一行代码都必须保证代码的鲁棒性。
import inspectimport itertoolsimport osimport tempfileimport torch.nn.functional as Fimport unittestimport numpy as npimport torchclass 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.003sOK
转载地址:http://kuwmi.baihongyu.com/