while t.tb_next:
t = t.tb_next
frame = t.tb_frame
- self.assertEqual(frame.f_lineno-frame.f_code.co_firstlineno, line)
+ if line is None:
+ self.assertEqual(frame.f_lineno, line)
+ else:
+ self.assertEqual(frame.f_lineno-frame.f_code.co_firstlineno, line)
def test_lineno_after_raise_simple(self):
def simple():
pass
self.lineno_after_raise(after_with, 2)
+ def test_missing_lineno_shows_as_none(self):
+ def f():
+ 1/0
+ self.lineno_after_raise(f, 1)
+ f.__code__ = f.__code__.replace(co_linetable=b'\x04\x80\xff\x80')
+ self.lineno_after_raise(f, None)
if __name__ == '__main__':
unittest.main()
--- /dev/null
+If the current position in a frame has no line number then set the f_lineno
+attribute to None, instead of -1, to conform to PEP 626. This should not
+normally be possible, but might occur in some unusual circumstances.
static PyObject *
frame_getlineno(PyFrameObject *f, void *closure)
{
- return PyLong_FromLong(PyFrame_GetLineNumber(f));
+ int lineno = PyFrame_GetLineNumber(f);
+ if (lineno < 0) {
+ Py_RETURN_NONE;
+ }
+ else {
+ return PyLong_FromLong(lineno);
+ }
}
static PyObject *