算法代码编写完毕后u需要进行几个测试,今天学习了一下ut测试,ut测试是测试算子代码的正确性,验证输入输出的结果是否和设计手册中的一致
def test_add_input(self):
add({"shape":(5,6,7), "dtype":"float32"},{"shape":(1,6,7), "dtype":"float32"},{"shape":(5,6,7), "dtype":"float32"},kernel_name="test_add_input")
比如这里,我用两个不同shape但是合法的input去测试,看代码能否通过测试。
具体的ut测试流程是:
- 打开testcases->ut->项目名->test项目名.py
- 在class Test_项目名_cce(unittest.TestCase)下编写各种测试函数,如上所列代码。
-
def main(): unittest.main() exit(0) if __name__ == "__main__": main()
使用unittest统一调用刚刚的代码,它会一次性完成所有的测试函数。
-
ut->算子名 右键run,完成后下方控制台会有一个测试报告。
另外
def tearDownClass(self):
# Must use the @classmethod decorator, run once after all tests have run