如何在python中获得测试方法的起始行号和结束行号？

Question

我目前正在尝试获取python文件中所有测试方法的第一行和最后一行。你知道吗

例如：

文件测试_电梯.py测试文件电梯.py. 你知道吗

测试_电梯.py

1 import random
2
3 from src.lift import Lift
4 from flaky import flaky
5
6
7 def test_upwards():
8         tester = Lift()
9         tester.upwards()
10        assert 1 == tester.curr_floor
11
12 def test_downwards():
13        tester = Lift()
14        tester.curr_floor = 1
15        tester.downwards()
16        assert 0 == tester.curr_floor
17        tester.downwards()
18        assert 0 == tester.curr_floor
19
...

我现在想得到test中每个测试方法的第一行和最后一行的编号_电梯.py，例如：

向上测试，7，10

向下测试，12，18

我已经试过使用确认测试.py，但没有成功。也许我忽略了什么？ 解决方案不一定必须是python。如果有人知道如何通过解析文件来解决这个问题，我可以接受。你知道吗

Answer 1

您可以使用inspect模块来完成此任务

import inspect
lines,line_start = inspect.getsourcelines(test_downwards)

print("Lines:",line_start,line_start+len(lines) - 1 )