1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
# This file is part of "austin" which is released under GPL.
#
# See file LICENCE or go to http://www.gnu.org/licenses/ for full license
# details.
#
# Austin is a Python frame stack sampler for CPython.
#
# Copyright (c) 2023 Gabriele N. Tornetta <phoenix1987@gmail.com>.
# All rights reserved.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from pathlib import Path
from test.utils import allpythons, austin, python, target
from austin.format.mojo import MojoFile, MojoFrame
@allpythons(min=(3, 11))
def test_mojo_column_data(py, tmp_path: Path):
datafile = tmp_path / "test_mojo_column.austin"
result = austin(
"-i", "100", "-o", str(datafile), *python(py), target("column.py"), mojo=True
)
assert result.returncode == 0, result.stderr or result.stdout
def strip(f):
return (f.scope.string.value, f.line, f.line_end, f.column, f.column_end)
with datafile.open("rb") as f:
frames = {
strip(_)
for _ in MojoFile(f).parse()
if isinstance(_, MojoFrame)
and _.filename.string.value.endswith("column.py")
}
assert frames & {
("<module>", 16, 19, 5, 2),
("<listcomp>", 16, 19, 5, 2),
("lazy", 6, 6, 9, 19),
("<listcomp>", 17, 17, 5, 17),
}
@allpythons(max=(3, 10))
def test_mojo_no_column_data(py, tmp_path: Path):
"""
Test that no other location information is present apart from the line
number for Python versions prior to 3.11.
"""
datafile = tmp_path / "test_mojo_column.austin"
result = austin(
"-i", "100", "-o", str(datafile), *python(py), target("column.py"), mojo=True
)
assert result.returncode == 0, result.stderr or result.stdout
def strip(f):
return
with datafile.open("rb") as f:
assert {
(e.line_end, e.column, e.column_end)
for e in MojoFile(f).parse()
if isinstance(e, MojoFrame)
} == {(0, 0, 0)}
|
