使用Python单元测试来测试数组中的结果

Question

我已经实现了计算年度抵押还款的函数，并且我想测试它是否会生成正确的结果。

我已经把阵列中的输入值的抵押贷款，如：

input_array([ 50000., 100000., 150000., ...]) 

并且在另一个阵列的相应结果：

output_array([ 3200.60, 6401.20, 9601.79, ...]) 

我想利用每一个元件在输入数组中作为我函数的输入并测试结果与output_array相同。

是否可以自动执行，而不是手动输入assertEqual函数中的抵押值和预期输出？

非常感谢。

Answer 1

您有两种选择。第一，只是比较全阵列：

expected_list = [1, 2, 3, 4] 
output_list = calc_mortgage([10, 20, 30, 42]) 
self.assertEqual(expected_list, output_list) 

第二，你可以比较每个元素：

expected_list = [1, 2, 3, 4] 
output_list = calc_mortgage([10, 20, 30, 42]) 
for pair in zip(expected_list, output_list): 
    self.assertEqual(pair[0], pair[1]) 

Answer 2

assertListEqual(map(f, input_array), output_array)