def test_simple_program_correct(self):
'''Can typecheck a simple program:
`int wildcat, animals[2];
float duck;
wildcat = 1;
duck = 1.0;
if(duck) {
animals[0] = wildcat;
animals[1] = 2;
} else {
animals[0] = 1;
}
duck = (duck + 2.5) * 33.0 / 2.0;
`
'''
given = ast.Block([
ast.Declaration('int', [
ast.ID('wildcat'),
ast.ArrayRef(ast.ID('animals'), ast.Integer(2))
]),
ast.Declaration('float', [ast.ID('duck')]),
ast.Assignment(ast.ID('wildcat'), ast.Integer(1)),
ast.Assignment(ast.ID('duck'), ast.Float(1.0)),
ast.IfStatement(
ast.ID('duck'),
ast.Block([
ast.Assignment(
ast.ArrayRef(ast.ID('animals'), ast.Integer(0)),
ast.ID('wildcat')),
ast.Assignment(
ast.ArrayRef(ast.ID('animals'), ast.Integer(1)),
ast.Integer(2))
]),
ast.Block([
ast.Assignment(
ast.ArrayRef(ast.ID('animals'), ast.Integer(0)),
ast.Integer(1))
])),
ast.Assignment(
ast.ID('duck'),
ast.BinOp(
'/',
ast.BinOp('*',
ast.BinOp('+', ast.ID('duck'), ast.Float(2.5)),
ast.Float(33.0)), ast.Float(2.0)))
])
report = typechecker.typecheck(given)
expected_errors = []
for (r, e) in zip_longest(report.get_errors(), expected_errors):
assert r == e
def test_compiler_parse(self):
'''Can parse a simple program:'''
given = '''
float duck, goose, birds[2];
int wildcat;
duck = 1.0;
goose = -1;
wildcat = 1;
birds[0] = duck;
if (duck) {
birds[1] = goose;
} else {
birds[1] = wildcat;
}
'''
result = compiler.parse(given)
expected = ast.Block([
ast.Declaration('float', [
ast.ID('duck'),
ast.ID('goose'),
ast.ArrayRef(ast.ID('birds'), ast.Integer(2))
]),
ast.Declaration('int', [ast.ID('wildcat')]),
ast.Assignment(ast.ID('duck'), ast.Float(1.0)),
ast.Assignment(ast.ID('goose'), ast.UnaryOp('-', ast.Integer(1))),
ast.Assignment(ast.ID('wildcat'), ast.Integer(1)),
ast.Assignment(ast.ArrayRef(ast.ID('birds'), ast.Integer(0)),
ast.ID('duck')),
ast.IfStatement(
ast.ID('duck'),
ast.Block([
ast.Assignment(
ast.ArrayRef(ast.ID('birds'), ast.Integer(1)),
ast.ID('goose'))
]),
ast.Block([
ast.Assignment(
ast.ArrayRef(ast.ID('birds'), ast.Integer(1)),
ast.ID('wildcat'))
])),
])
assert result == expected
def test_compiler_typecheck(self):
'''Can find errors in a simple program:'''
given = '''
float duck, goose, birds[2];
int wildcat;
duck = 1.0;
goose = -1;
wildcat = 1;
birds[0] = duck;
if (duck) {
birds[1] = goose;
} else {
birds[1] = wildcat;
}
'''
result = compiler.typecheck(given)
expected_errors = [
TypecheckerError(
ast.Assignment(ast.ID('goose'),
ast.UnaryOp('-', ast.Integer(1))), ),
TypecheckerError(
ast.Assignment(ast.ArrayRef(ast.ID('birds'), ast.Integer(1)),
ast.ID('wildcat')))
]
for (r, e) in zip_longest(result.get_errors(), expected_errors):
assert r == e
def test_arrayref_expr_correct(self):
'''No error with given program:
`int x[3];
x[0] = 1;`'''
given = ast.Block([
ast.Declaration('int',
[ast.ArrayRef(ast.ID('x'), ast.Integer(3))]),
ast.Assignment(ast.ArrayRef(ast.ID('x'), ast.Integer(0)),
ast.Integer(1))
])
report = typechecker.typecheck(given)
expected_errors = []
for (r, e) in zip_longest(report.get_errors(), expected_errors):
assert r == e
def test_arrayref_expression(self):
'''Array reference is an expression: `arr[1];`'''
given = iter([
Token('ID', 'arr'),
Token('LBRACE', '['),
Token('INTEGER', '1'),
Token('RBRACE', ']'),
Token('SEMI', ';')
])
result = parser.parse(given)
expected = ast.Block([
ast.Statement(
ast.ArrayRef(ast.ID('arr'), ast.Integer(1))
)
])
assert result == expected
def test_simple_program(self):
'''Can parse a simple program
`int x[3], duck, goose, wildcat;
duck = 1;
goose = 2;
x[0] = duck;
x[duck] = goose;
if (duck + x[duck]) {
wildcat = duck;
} else {
wildcat = goose;
}
x[2] = wildcat;`
'''
given = iter([
Token('INT_TYPE', 'int'),
Token('ID', 'x'), Token('LBRACE', '['), Token('INTEGER', '3'), Token('RBRACE', ']'), Token('COMMA', ','),
Token('ID', 'duck'), Token('COMMA', ','),
Token('ID', 'goose'), Token('COMMA', ','),
Token('ID', 'wildcat'), Token('SEMI', ';'),
Token('ID', 'duck'), Token('EQUAL', '='), Token('INTEGER', '1'), Token('SEMI', ';'),
Token('ID', 'goose'), Token('EQUAL', '='), Token('INTEGER', '2'), Token('SEMI', ';'),
Token('ID', 'x'), Token('LBRACE', '['), Token('INTEGER', '0'), Token('RBRACE', ']'), Token('EQUAL', '='), Token('ID', 'duck'), Token('SEMI', ';'),
Token('ID', 'x'), Token('LBRACE', '['), Token('ID', 'duck'), Token('RBRACE', ']'), Token('EQUAL', '='), Token('ID', 'goose'), Token('SEMI', ';'),
Token('IF', 'if'), Token('LPAREN', '('), Token('ID', 'duck'), Token('PLUS', '+'), Token('ID', 'x'), Token('LBRACE', '['), Token('ID', 'duck'), Token('RBRACE', ']'), Token('RPAREN', ')'),
Token('LCURLY', '{'), Token('ID', 'wildcat'), Token('EQUAL', '='), Token('ID', 'duck'), Token('SEMI', ';'), Token('RCURLY', '}'),
Token('ELSE', 'else'), Token('LCURLY', '{'), Token('ID', 'wildcat'), Token('EQUAL', '='), Token('ID', 'goose'), Token('SEMI', ';'), Token('RCURLY', '}'),
Token('ID', 'x'), Token('LBRACE', '['), Token('INTEGER', '2'), Token('RBRACE', ']'), Token('EQUAL', '='), Token('ID', 'wildcat'), Token('SEMI', ';')
])
expected = ast.Block([
ast.Declaration(
'int',
[ast.ArrayRef(ast.ID('x'), ast.Integer(3)), ast.ID('duck'), ast.ID('goose'), ast.ID('wildcat')]
),
ast.Assignment(ast.ID('duck'), ast.Integer(1)),
ast.Assignment(ast.ID('goose'), ast.Integer(2)),
ast.Assignment(ast.ArrayRef(ast.ID('x'), ast.Integer(0)), ast.ID('duck')),
ast.Assignment(ast.ArrayRef(ast.ID('x'), ast.ID('duck')), ast.ID('goose')),
ast.IfStatement(
ast.BinOp('+', ast.ID('duck'), ast.ArrayRef(ast.ID('x'), ast.ID('duck'))),
ast.Block([
ast.Assignment(ast.ID('wildcat'), ast.ID('duck'))
]),
ast.Block([
ast.Assignment(ast.ID('wildcat'), ast.ID('goose'))
])
),
ast.Assignment(ast.ArrayRef(ast.ID('x'), ast.Integer(2)), ast.ID('wildcat')),
])
result = parser.parse(given)
assert expected == result
def variable_arrayref(s):
return ast.ArrayRef(s[0], s[2])