我希望能够使用来自一组边缘情况和正常值的元组来测试一个函数。例如,在测试一个在给定三个长度形成有效三角形时返回 true 的函数时,我会遇到特定情况,负数、小数、大数、接近溢出的值等等。然后生成这些值的组合,有或没有重复,以获得一组测试数据。
> (inf,0,-1), (5,10,1000), (10,5,5), (0,-1,5), (1000,inf,inf),
>
> ...
我希望能够使用来自一组边缘情况和正常值的元组来测试一个函数。例如,在测试一个在给定三个长度形成有效三角形时返回 true 的函数时,我会遇到特定情况,负数、小数、大数、接近溢出的值等等。然后生成这些值的组合,有或没有重复,以获得一组测试数据。
> (inf,0,-1), (5,10,1000), (10,5,5), (0,-1,5), (1000,inf,inf),
>
> ...
使用全新的 Python 2.6,您可以使用 itertools 模块返回 iterables 的笛卡尔积的标准解决方案:
import itertools
print list(itertools.product([1,2,3], [4,5,6]))
[(1, 4), (1, 5), (1, 6),
(2, 4), (2, 5), (2, 6),
(3, 4), (3, 5), (3, 6)]
您可以提供“重复”参数以使用可迭代和自身执行产品:
print list(itertools.product([1,2], repeat=3))
[(1, 1, 1), (1, 1, 2), (1, 2, 1), (1, 2, 2),
(2, 1, 1), (2, 1, 2), (2, 2, 1), (2, 2, 2)]
你也可以用组合来调整一些东西:
print list(itertools.combinations('123', 2))
[('1', '2'), ('1', '3'), ('2', '3')]
如果顺序很重要,则有排列:
print list(itertools.permutations([1,2,3,4], 2))
[(1, 2), (1, 3), (1, 4),
(2, 1), (2, 3), (2, 4),
(3, 1), (3, 2), (3, 4),
(4, 1), (4, 2), (4, 3)]
当然,所有那些很酷的东西并不完全相同,但你可以以某种方式使用它们来解决你的问题。
请记住,您可以使用 list()、tuple() 和 set() 将元组或列表转换为集合,反之亦然。