查找元素
选择合适的数据结构可以显著提升代码性能。例如,若要在数据中查找某个元素,您可能会在 list 和 set 之间做选择。在本练习中,您将使用 pytest 实现性能测试,比较 in 运算符分别用于两种数据结构 list 和 set 时的速度。pytest 包已被导入。
本练习是课程的一部分
Python 测试入门
练习说明
- 将
benchmark作为参数传入测试函数。 - 然后在测试函数中调用
benchmark(),并将find()作为第 1 个参数传入。
交互式实操练习
通过完成这段示例代码来试试这个练习。
def create_list():
return [i for i in range(1000)]
def create_set():
return set([i for i in range(1000)])
def find(it, el=50):
return el in it
# Write the performance test for a list
def test_list(____):
____(____, ____)
# Write the performance test for a set
def test_set(____):
____(____, ____)