PythonBuddy/pypy/rpython/flowspace/test/test_checkgraph.py at restrictedpython · BoragoCode/PythonBuddy · GitHub

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
79
80
81
82
83
84
85
86
from rpython.flowspace.model import *
import py

def test_mingraph():
    g = FunctionGraph("g", Block([]))
    g.startblock.closeblock(Link([Constant(1)], g.returnblock))
    checkgraph(g)

def template():
    g = FunctionGraph("g", Block([]))
    g.startblock.closeblock(Link([Constant(1)], g.returnblock))
    checkgraph(g)
    py.test.raises(AssertionError, checkgraph, g)


def test_exitlessblocknotexitblock():
    g = FunctionGraph("g", Block([]))
    py.test.raises(AssertionError, checkgraph, g)


def test_nonvariableinputarg():
    b = Block([Constant(1)])
    g = FunctionGraph("g", b)
    g.startblock.closeblock(Link([Constant(1)], g.returnblock))

    py.test.raises(AssertionError, checkgraph, g)

def test_multiplydefinedvars():
    v = Variable()
    g = FunctionGraph("g", Block([v, v]))
    g.startblock.closeblock(Link([v], g.returnblock))
    py.test.raises(AssertionError, checkgraph, g)

    v = Variable()
    b = Block([v])
    b.operations.append(SpaceOperation("add", [Constant(1), Constant(2)], v))
    g = FunctionGraph("g", b)
    g.startblock.closeblock(Link([v], g.returnblock))

    py.test.raises(AssertionError, checkgraph, g)

def test_varinmorethanoneblock():
    v = Variable()
    g = FunctionGraph("g", Block([]))
    g.startblock.operations.append(SpaceOperation("pos", [Constant(1)], v))
    b = Block([v])
    g.startblock.closeblock(Link([v], b))
    b.closeblock(Link([v], g.returnblock))
    py.test.raises(AssertionError, checkgraph, g)

def test_useundefinedvar():
    v = Variable()
    g = FunctionGraph("g", Block([]))
    g.startblock.closeblock(Link([v], g.returnblock))
    py.test.raises(AssertionError, checkgraph, g)

    v = Variable()
    g = FunctionGraph("g", Block([]))
    g.startblock.exitswitch = v
    g.startblock.closeblock(Link([Constant(1)], g.returnblock))
    py.test.raises(AssertionError, checkgraph, g)

def test_invalid_arg():
    v = Variable()
    g = FunctionGraph("g", Block([]))
    g.startblock.operations.append(SpaceOperation("pos", [1], v))
    g.startblock.closeblock(Link([v], g.returnblock))
    py.test.raises(AssertionError, checkgraph, g)

def test_invalid_links():
    g = FunctionGraph("g", Block([]))
    g.startblock.closeblock(Link([Constant(1)], g.returnblock), Link([Constant(1)], g.returnblock))
    py.test.raises(AssertionError, checkgraph, g)

    v = Variable()
    g = FunctionGraph("g", Block([v]))
    g.startblock.exitswitch = v
    g.startblock.closeblock(Link([Constant(1)], g.returnblock, True),
                            Link([Constant(1)], g.returnblock, True))
    py.test.raises(AssertionError, checkgraph, g)

    v = Variable()
    g = FunctionGraph("g", Block([v]))
    g.startblock.exitswitch = v
    g.startblock.closeblock(Link([Constant(1)], g.returnblock))
    py.test.raises(AssertionError, checkgraph, g)