如何在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


def line_number():
    return inspect.currentframe().f_back.f_lineno


print "This is line number:", line_number()
this = line_number()
print "This is line number:", line_number()
print "Variable 'this' at line: ", this
line_number()

输出：

This is line number: 8
This is line number: 10
Variable 'this' at line: 9

Answer 2

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

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

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

Answer 3

或者，在没有任何附加模块（但有一些Python内部构件）的情况下：

>>> def thing():
...  a = 5
...  b = a + 4
...  return a + b * 2
... 
>>> thing.__code__.co_firstlineno  # self-explanatory
1
>>> list(thing.__code__.co_lnotab)
[0, 1, 4, 1, 8, 1]
>>> thing.__code__.co_firstlineno + sum(line_increment for i, line_increment in enumerate(thing.__code__.co_lnotab) if i % 2)
4

这样就有了：函数从第1行（thing.__code__.co_firstlineno）到第4行。你知道吗

dis模块证明了这一点（第一列中的数字是行号）：

>>> import dis
>>> dis.dis(thing)
  2           0 LOAD_CONST               1 (5)
              2 STORE_FAST               0 (a)

  3           4 LOAD_FAST                0 (a)
              6 LOAD_CONST               2 (4)
              8 BINARY_ADD
             10 STORE_FAST               1 (b)

  4          12 LOAD_FAST                0 (a)
             14 LOAD_FAST                1 (b)
             16 LOAD_CONST               3 (2)
             18 BINARY_MULTIPLY
             20 BINARY_ADD
             22 RETURN_VALUE

注意最后一个数字是4，这是函数最后一行的数字。你知道吗

更多关于co_lnotab结构的信息可以在here和here中找到。你知道吗

---

测试程序

def span_of_function(func):
  start = func.__code__.co_firstlineno
  end = start + sum(line_increment for i, line_increment in enumerate(func.__code__.co_lnotab) if i % 2)

  return start, end

def hello(name):
  print(f'Hello, {name}!')
  return 5

print(span_of_function(span_of_function))
print(span_of_function(hello))

输出：

$ python3 test.py
(1, 5)
(7, 9)