FilesExpand file tree

main

/

test_propagation_phase2.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

72 lines (59 loc) · 2.58 KB

main

/

test_propagation_phase2.py

Copy path

Top

File metadata and controls

• Code
• Blame

72 lines (59 loc) · 2.58 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

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

"""Tests for Phase 2 propagation primitives."""

from codepathfinder.propagation import PropagationPrimitive, PropagationType, propagates

class TestPhase2Primitives:

"""Test Phase 2 string operation primitives."""

def test_string_concat_returns_primitive(self):

"""Test propagates.string_concat() returns PropagationPrimitive."""

prim = propagates.string_concat()

assert isinstance(prim, PropagationPrimitive)

assert prim.type == PropagationType.STRING_CONCAT

def test_string_concat_ir(self):

"""Test propagates.string_concat() serializes correctly."""

prim = propagates.string_concat()

ir = prim.to_ir()

assert ir["type"] == "string_concat"

assert ir["metadata"] == {}

def test_string_format_returns_primitive(self):

"""Test propagates.string_format() returns PropagationPrimitive."""

prim = propagates.string_format()

assert isinstance(prim, PropagationPrimitive)

assert prim.type == PropagationType.STRING_FORMAT

def test_string_format_ir(self):

"""Test propagates.string_format() serializes correctly."""

prim = propagates.string_format()

ir = prim.to_ir()

assert ir["type"] == "string_format"

assert ir["metadata"] == {}

def test_string_concat_repr(self):

"""Test string_concat __repr__."""

prim = propagates.string_concat()

assert repr(prim) == "propagates.string_concat()"

def test_string_format_repr(self):

"""Test string_format __repr__."""

prim = propagates.string_format()

assert repr(prim) == "propagates.string_format()"

class TestPhase2Integration:

"""Integration tests for Phase 2 primitives."""

def test_all_phase2_primitives(self):

"""Test all Phase 2 primitives can be used together."""

prims = [

propagates.string_concat(),

propagates.string_format(),

]

assert len(prims) == 2

types = [p.type for p in prims]

assert PropagationType.STRING_CONCAT in types

assert PropagationType.STRING_FORMAT in types

def test_phase1_and_phase2_together(self):

"""Test Phase 1 and Phase 2 primitives work together."""

prims = [

propagates.assignment(),

propagates.function_args(),

propagates.function_returns(),

propagates.string_concat(),

propagates.string_format(),

]

assert len(prims) == 5

# All primitives should serialize to IR

ir_list = [p.to_ir() for p in prims]

assert len(ir_list) == 5