def test_parser9():
assert str(ArrowParser().parse("for var i = 0; i < 20; i = i + 1 {print_int32(i) continue}")) == '''
1:Stmts
4:For
1:DeclExpr
2:ShortDecl
0:Name,i
0:Int,0
1:BooleanExpr
2:<
0:Symbol,i
0:Int,20
1:UpdateExpr
2:AssignStmt
0:Name,i
2:+
0:Symbol,i
0:Int,1
2:Block
2:Call
0:Symbol,print_int32
1:Params
0:Symbol,i
0:Continue
'''.strip()
def test_parser1():
assert str(ArrowParser().parse("var x = 0")) == '''
1:Stmts
2:ShortDecl
0:Name,x
0:Int,0
'''.strip()
def test_parser2():
assert str(ArrowParser().parse("x = 1")) == '''
1:Stmts
2:AssignStmt
0:Name,x
0:Int,1
'''.strip()
def test_checker4():
assert str(ArrowTypechecker(ArrowParser().parse('''
var x = 0
if true {
x = 1
} else {
x = 2
}
''')).typecheck()) == """
2:Stmts:unit
2:ShortDecl:unit
0:Name,x:int32
0:Int,0:int32
3:If:unit
1:BooleanExpr:boolean
0:Boolean,true:boolean
1:Block:unit
2:AssignStmt:unit
0:Name,x:int32
0:Int,1:int32
1:ElseIf:unit
1:Block:unit
2:AssignStmt:unit
0:Name,x:int32
0:Int,2:int32
""".strip()
def test_checker6():
assert str(ArrowTypechecker(ArrowParser().parse("print_int32(0)")).typecheck()) == '''
1:Stmts:unit
2:Call:unit
0:Symbol,print_int32:fn(int32)->unit
1:Params:(int32)
0:Int,0:int32
'''.strip()
def test_parser8():
assert str(ArrowParser().parse("print_int32(i)")) == '''
1:Stmts
2:Call
0:Symbol,print_int32
1:Params
0:Symbol,i
'''.strip()
def test_lexer_string():
assert str(ArrowParser().parse('''
var x = "a string with escaped \\" and \\\\"
''')) == '''
1:Stmts
2:ShortDecl
0:Name,x
0:String,a string with escaped \\" and \\
'''.strip()
def test_parser5():
assert str(ArrowParser().parse("var x = 1 + 1")) == '''
1:Stmts
2:ShortDecl
0:Name,x
2:+
0:Int,1
0:Int,1
'''.strip()
def test_parser4():
assert str(ArrowParser().parse("var x = 1 - 1.0")) == '''
1:Stmts
2:ShortDecl
0:Name,x
2:-
0:Int,1
0:Float,1.0
'''.strip()
def test_parser10():
assert str(ArrowParser().parse("while(true){break}")) == '''
1:Stmts
4:For
0:DeclExpr
1:BooleanExpr
0:Boolean,true
0:UpdateExpr
1:Block
0:Break
'''.strip()
def test_parser6():
assert str(ArrowParser().parse("var x int32 = 1*1")) == '''
1:Stmts
3:Decl
0:Name,x
1:Type
0:TypeName,int32
2:*
0:Int,1
0:Int,1
'''.strip()
def test_checker7():
assert str(ArrowTypechecker(ArrowParser().parse("var x float32 = float32(1)")).typecheck()) == '''
1:Stmts:unit
3:Decl:unit
0:Name,x:float32
1:Type:float32
0:TypeName,float32:float32
2:Cast:float32
0:Symbol,float32:fn(int32)->float32
1:Params:(int32)
0:Int,1:int32
'''.strip()
def test_lexer_comments():
assert str(ArrowParser().parse('''
/*line comment \* escape sequences \/ \*/
newlines
end comment */
var x = 0
//line comment
''')) == '''
1:Stmts
2:ShortDecl
0:Name,x
0:Int,0
'''.strip()
def test_parser7():
assert str(ArrowParser().parse("func f (i int32) int32 { return 5}")) == '''
1:Stmts
4:FuncDef
0:Name,f
1:ParamDecls
2:ParamDecl
0:Name,i
1:Type
0:TypeName,int32
1:ReturnType
1:Type
0:TypeName,int32
1:Block
1:Return
0:Int,5
'''.strip()
def test_parser3():
assert str(ArrowParser().parse("if x == 0 {y = 1} else {y = 0}")) == '''
1:Stmts
3:If
1:BooleanExpr
2:==
0:Symbol,x
0:Int,0
1:Block
2:AssignStmt
0:Name,y
0:Int,1
1:ElseIf
1:Block
2:AssignStmt
0:Name,y
0:Int,0
'''.strip()
def test_checker2():
assert str(ArrowTypechecker(ArrowParser().parse("var x = 1 * 2 + 3 * (-1 + 7) / 2")).typecheck()) == '''
1:Stmts:unit
2:ShortDecl:unit
0:Name,x:int32
2:+:int32
2:*:int32
0:Int,1:int32
0:Int,2:int32
2:/:int32
2:*:int32
0:Int,3:int32
2:+:int32
1:Negate:int32
0:Int,1:int32
0:Int,7:int32
0:Int,2:int32
'''.strip()
def test_checker3():
assert str(ArrowTypechecker(ArrowParser().parse("var x float32 = 1.0 * 2.0 + 3.0 * (-1.0 + 7.0) / 2.0")).typecheck()) == '''
1:Stmts:unit
3:Decl:unit
0:Name,x:float32
1:Type:float32
0:TypeName,float32:float32
2:+:float32
2:*:float32
0:Float,1.0:float32
0:Float,2.0:float32
2:/:float32
2:*:float32
0:Float,3.0:float32
2:+:float32
1:Negate:float32
0:Float,1.0:float32
0:Float,7.0:float32
0:Float,2.0:float32
'''.strip()
def get_il(src):
return ILGenerator(
ArrowTypechecker(
ArrowParser().parse(src)
).typecheck()
).get_prettyprint()
def test_parser11():
assert str(ArrowParser().parse('''
var x = 1
var x = 2
func foo() { return }
func bar() { return }
func baz() { return }
if x == y && (y == z || z == c) {
foo()
} else if (y < z) || (z > x && x != y) {
bar()
} else {
baz()
}''')) == """
6:Stmts
2:ShortDecl
0:Name,x
0:Int,1
2:ShortDecl
0:Name,x
0:Int,2
4:FuncDef
0:Name,foo
0:ParamDecls
1:ReturnType
1:Type
0:TypeName,unit
1:Block
0:Return
4:FuncDef
0:Name,bar
0:ParamDecls
1:ReturnType
1:Type
0:TypeName,unit
1:Block
0:Return
4:FuncDef
0:Name,baz
0:ParamDecls
1:ReturnType
1:Type
0:TypeName,unit
1:Block
0:Return
3:If
1:BooleanExpr
2:And
2:==
0:Symbol,x
0:Symbol,y
2:Or
2:==
0:Symbol,y
0:Symbol,z
2:==
0:Symbol,z
0:Symbol,c
1:Block
2:Call
0:Symbol,foo
0:Params
1:ElseIf
3:If
1:BooleanExpr
2:Or
2:<
0:Symbol,y
0:Symbol,z
2:And
2:>
0:Symbol,z
0:Symbol,x
2:!=
0:Symbol,x
0:Symbol,y
1:Block
2:Call
0:Symbol,bar
0:Params
1:ElseIf
1:Block
2:Call
0:Symbol,baz
0:Params
""".strip()
def test_checker5():
assert str(ArrowTypechecker(ArrowParser().parse('''
func fib(x int32) int32 {
if x <= 0 {
return 0
}
var prev = 0
var cur = 1
for var i = 1; i < x; i = i + 1 {
var next = cur + prev
prev = cur
cur = next
}
return cur
}
print_int32(fib(10))
''')).typecheck()) == """
2:Stmts:unit
4:FuncDef:unit
0:Name,fib:fn(int32)->int32
1:ParamDecls:(int32)
2:ParamDecl:int32
0:Name,x:int32
1:Type:int32
0:TypeName,int32:int32
1:ReturnType:int32
1:Type:int32
0:TypeName,int32:int32
5:Block:unit
3:If:unit
1:BooleanExpr:boolean
2:<=:boolean
0:Symbol,x:int32
0:Int,0:int32
1:Block:unit
1:Return:unit
0:Int,0:int32
0:ElseIf:unit
2:ShortDecl:unit
0:Name,prev:int32
0:Int,0:int32
2:ShortDecl:unit
0:Name,cur:int32
0:Int,1:int32
4:For:unit
1:DeclExpr:unit
2:ShortDecl:unit
0:Name,i:int32
0:Int,1:int32
1:BooleanExpr:boolean
2:<:boolean
0:Symbol,i:int32
0:Symbol,x:int32
1:UpdateExpr:unit
2:AssignStmt:unit
0:Name,i:int32
2:+:int32
0:Symbol,i:int32
0:Int,1:int32
3:Block:unit
2:ShortDecl:unit
0:Name,next:int32
2:+:int32
0:Symbol,cur:int32
0:Symbol,prev:int32
2:AssignStmt:unit
0:Name,prev:int32
0:Symbol,cur:int32
2:AssignStmt:unit
0:Name,cur:int32
0:Symbol,next:int32
1:Return:unit
0:Symbol,cur:int32
2:Call:unit
0:Symbol,print_int32:fn(int32)->unit
1:Params:(int32)
2:Call:int32
0:Symbol,fib:fn(int32)->int32
1:Params:(int32)
0:Int,10:int32
""".strip()