FilesExpand file tree

main

/

test_bytecode.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

44 lines (34 loc) · 1.44 KB

main

/

test_bytecode.py

Copy path

Top

File metadata and controls

• Code
• Blame

44 lines (34 loc) · 1.44 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0

# For details: https://github.com/coveragepy/coveragepy/blob/main/NOTICE.txt

"""Tests for coverage.py's bytecode analysis."""

from __future__ import annotations

import dis

from textwrap import dedent

from coverage import env

from coverage.bytecode import ByteParser, op_set

from tests.coveragetest import CoverageTest

class BytecodeTest(CoverageTest):

"""Tests for bytecode.py"""

def test_code_objects(self) -> None:

bp = ByteParser(

text=dedent("""\

def f(x):

def g(y):

return {z for z in range(10)}

def j():

return [z for z in range(10)]

return g(x)

def h(x):

return x+1

"""),

)

objs = list(bp.code_objects())

expected = {"<module>", "f", "g", "j", "h"}

if env.PYVERSION < (3, 12) or env.PYPY:

# Comprehensions were compiled as implicit functions in earlier

# versions of CPython. PyPy still compiles them that way.

expected.update({"<setcomp>", "<listcomp>"})

assert {c.co_name for c in objs} == expected

def test_op_set(self) -> None:

opcodes = op_set("LOAD_CONST", "NON_EXISTENT_OPCODE", "RETURN_VALUE")

assert opcodes == {dis.opmap["LOAD_CONST"], dis.opmap["RETURN_VALUE"]}