def test_if_else_statement():
source = """
import System::Output;
export function Main(): void {
if (5 < 10) {
println(value=1);
} else {
println(value=0);
}
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[],
functions=[
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[
If(
condition=BinaryOperation(
operator=BinaryOperator.LESS_THAN,
left=Number(5),
right=Number(10),
),
then_statements=[
FunctionCall(
name="println",
arguments=[
Argument(name="value",
value=(Number(1)))
],
)
],
else_statements=[
FunctionCall(
name="println",
arguments=[
Argument(name="value",
value=(Number(0)))
],
)
],
)
],
)
],
),
)
def test_simple_expression():
source = """
import System::Output;
export function Main(): void {
println(value=1 + 2 * 3);
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[],
functions=[
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[
FunctionCall(
name="println",
arguments=[
Argument(
name="value",
value=BinaryOperation(
operator=BinaryOperator.ADD,
left=Number(value=1),
right=BinaryOperation(
operator=BinaryOperator.MULTIPLY,
left=Number(value=2),
right=Number(value=3),
),
),
)
],
)
],
)
],
),
)
def test_simple_assignment():
source = """
import System::Output;
export function Main(): void {
let a = 5;
let b = 10;
println(value=a + b);
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[],
functions=[
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[
Let(name="a", value=Number(5)),
Let(name="b", value=Number(10)),
FunctionCall(
name="println",
arguments=[
Argument(
name="value",
value=BinaryOperation(
operator=BinaryOperator.ADD,
left=Variable(name="a"),
right=Variable(name="b"),
),
)
],
),
],
)
],
),
)
def test_data_vector_with_simple_function():
source = """
import System::Output;
export data Vector(x: i32, y: i32) {
function length(): i32 {
42;
}
}
export function Main(): void {
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[
DataDef(
name="Vector",
implements=[],
type=types.Placeholder(text="Vector"),
is_exported=True,
params=[
Param(name="x", type=types.Placeholder(text="i32")),
Param(name="y", type=types.Placeholder(text="i32")),
],
functions=[
Function(
name="length",
is_exported=False,
params=[],
type=types.Placeholder(text="i32"),
body=[Number(value=42)],
)
],
)
],
functions=[
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[],
)
],
),
)
def parse_minimal(self):
# factor => number
if self.match('new'):
self.advance()
return NewObject(self.previous_token["data"])
if self.match('string'):
return String(self.previous_token["data"][1:-1])
if self.match('number'):
return Number(self.previous_token["data"])
if self.match("lsquare"):
values = []
while values == [] or self.check("comma"):
if self.check("comma"):
self.advance()
val = self.parse_expression()
values.append(val)
self.expect("rsquare")
return Array(values)
# factor => id
if self.match("id"):
id = self.previous_token["data"]
if self.match("lparen"):
if self.match("rparen"):
return FunctionCall(id, [])
else:
arg = self.parse_expression()
args = [arg]
while self.match("comma"):
arg = self.parse_expression()
args.append(arg)
self.match("rparen")
return FunctionCall(id, args)
else:
return Variable(id)
# factor => ( exp )
if self.match("lparen"):
exp = self.parse_expression()
if exp is not None:
self.expect("rparen")
return exp
def test_function_call_with_no_arguments():
source = """
import System::Output;
function sayNumber(): void {
println(value=1337);
}
export function Main(): void {
sayNumber();
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[],
functions=[
Function(
name="sayNumber",
is_exported=False,
params=[],
type=types.Placeholder(text="void"),
body=[
FunctionCall(
name="println",
arguments=[
Argument(name="value", value=Number(1337))
],
)
],
),
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[FunctionCall(name="sayNumber", arguments=[])],
),
],
),
)
def test_trait_with_single_implementation():
source = """
trait List {
function length(self: List): i32;
function sum(self: List): i32;
}
data EmptyList() implements List {
function length(self: List): i32 {
0;
}
function sum(self: List): i32 {
0;
}
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[],
traits=[
Trait(
name="List",
is_exported=False,
type=types.Placeholder("List"),
functions=[
Function(
name="length",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=None,
),
Function(
name="sum",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=None,
),
],
)
],
data_defs=[
DataDef(
name="EmptyList",
is_exported=False,
implements=["List"],
params=[],
functions=[
Function(
name="length",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=[Number(value=0)],
),
Function(
name="sum",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=[Number(value=0)],
),
],
type=types.Placeholder(text="EmptyList"),
)
],
functions=[],
),
)
def test_data_vector_with_simple_function_call():
source = """
import System::Output;
export data Vector(x: i32, y: i32) {
function length(): i32 {
42;
}
}
export function Main(): void {
let vector = Vector(x=10, y=20);
println(value=vector.length());
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[
DataDef(
name="Vector",
implements=[],
type=types.Placeholder(text="Vector"),
is_exported=True,
params=[
Param(name="x", type=types.Placeholder(text="i32")),
Param(name="y", type=types.Placeholder(text="i32")),
],
functions=[
Function(
name="length",
is_exported=False,
params=[],
type=types.Placeholder(text="i32"),
body=[Number(value=42)],
)
],
)
],
functions=[
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[
Let(
name="vector",
value=FunctionCall(
name="Vector",
arguments=[
Argument(name="x", value=Number(value=10)),
Argument(name="y", value=Number(value=20)),
],
),
),
FunctionCall(
name="println",
arguments=[
Argument(
name="value",
value=FunctionCall(name="vector.length",
arguments=[]),
)
],
),
],
)
],
),
)
def test_factorial():
source = """
import System::Output;
function factorial(n: i32): i32 {
if (n == 1) {
return 1;
} else {
return n * factorial(n=n-1);
}
}
export function Main(): void {
println(value=factorial(n=5));
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[],
functions=[
Function(
name="factorial",
is_exported=False,
params=[
Param(name="n", type=types.Placeholder(text="i32"))
],
type=types.Placeholder(text="i32"),
body=[
If(
condition=BinaryOperation(
operator=BinaryOperator.EQUALS,
left=Variable("n"),
right=Number(1),
),
then_statements=[Return(value=Number(value=1))],
else_statements=[
Return(value=BinaryOperation(
operator=BinaryOperator.MULTIPLY,
left=Variable(name="n"),
right=FunctionCall(
name="factorial",
arguments=[
Argument(
name="n",
value=BinaryOperation(
operator=BinaryOperator.
SUBTRACT,
left=Variable(name="n"),
right=Number(value=1),
),
)
],
),
))
],
)
],
),
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[
FunctionCall(
name="println",
arguments=[
Argument(
name="value",
value=FunctionCall(
name="factorial",
arguments=[
Argument(name="n",
value=(Number(5)))
],
),
)
],
)
],
),
],
),
)
def test_function_call_with_arguments():
source = """
import System::Output;
function add(x: i32, y: i32): i32 {
x + y;
}
export function Main(): void {
println(value=add(a=5, b=15));
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[],
functions=[
Function(
name="add",
is_exported=False,
params=[
Param(name="x", type=types.Placeholder(text="i32")),
Param(name="y", type=types.Placeholder(text="i32")),
],
type=types.Placeholder(text="i32"),
body=[
BinaryOperation(
operator=BinaryOperator.ADD,
left=Variable("x"),
right=Variable("y"),
)
],
),
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[
FunctionCall(
name="println",
arguments=[
Argument(
name="value",
value=(FunctionCall(
name="add",
arguments=[
Argument(name="a",
value=Number(value=5)),
Argument(name="b",
value=Number(value=15)),
],
)),
)
],
)
],
),
],
),
)
def test_trait_with_multiple_implementations():
source = """
trait List {
function length(self: List): i32;
function sum(self: List): i32;
}
data EmptyList() implements List {
function length(self: List): i32 {
0;
}
function sum(self: List): i32 {
0;
}
}
data Cons(head: i32, tail: List) implements List {
function length(self: List): i32 {
1 + self.tail.length();
}
function sum(self: List): i32 {
self.head + self.tail.sum();
}
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[],
traits=[
Trait(
name="List",
is_exported=False,
functions=[
Function(
name="length",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=None,
),
Function(
name="sum",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=None,
),
],
type=types.Placeholder(text="List"),
)
],
data_defs=[
DataDef(
name="EmptyList",
is_exported=False,
implements=["List"],
params=[],
functions=[
Function(
name="length",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=[Number(value=0)],
),
Function(
name="sum",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=[Number(value=0)],
),
],
type=types.Placeholder(text="EmptyList"),
),
DataDef(
name="Cons",
is_exported=False,
implements=["List"],
params=[
Param(name="head", type=types.Placeholder(text="i32")),
Param(name="tail",
type=types.Placeholder(text="List")),
],
functions=[
Function(
name="length",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=[
BinaryOperation(
operator=BinaryOperator.ADD,
left=Number(value=1),
right=FunctionCall(name="self.tail.length",
arguments=[]),
)
],
),
Function(
name="sum",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="List"))
],
type=types.Placeholder(text="i32"),
body=[
BinaryOperation(
operator=BinaryOperator.ADD,
left=MemberAccess(
value=Variable(name="self"),
member="head"),
right=FunctionCall(name="self.tail.sum",
arguments=[]),
)
],
),
],
type=types.Placeholder(text="Cons"),
),
],
functions=[],
),
)
def number(state, p):
return Number(self.cg, state, p[0].value)