def visitData(self, ctx: BugParser.DataContext):
params = []
if ctx.dataList():
for param in ctx.dataList().params:
var_name = param.variableName().getText()
type_ = param.typeName().getText()
params.append(Param(type=types.Placeholder(type_), name=var_name))
functions = []
if ctx.functionDef():
for function in ctx.functionDef():
functions.append(self.visit(function))
implements = []
if ctx.traitName():
implements.append(ctx.traitName().getText())
name = ctx.dataName().getText()
return DataDef(
name=name,
implements=implements,
type=types.Placeholder(text=name),
is_exported=ctx.EXPORT() is not None,
params=params,
functions=functions,
)
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 test_trait_with_functions():
source = """
trait List {
function length(self: List): i32;
function sum(self: List): i32;
}
"""
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=[],
functions=[],
),
)
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 visitTraitFunctionDef(self, ctx: BugParser.TraitFunctionDefContext):
function_name = ctx.traitFunctionName().getText()
return_type = ctx.returnTypeName().getText()
params = []
if ctx.parameterList():
for param in ctx.parameterList().params:
var_name = param.variableName().getText()
type_ = param.typeName().getText()
params.append(Param(type=types.Placeholder(type_), name=var_name))
return Function(
name=function_name,
is_exported=False,
params=params,
type=types.Placeholder(text=return_type),
body=None,
)
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 visitFunctionDef(self, ctx: BugParser.FunctionDefContext):
function_name = ctx.functionName().getText()
return_type = ctx.returnTypeName().getText()
params = []
if ctx.parameterList():
for param in ctx.parameterList().params:
var_name = param.variableName().getText()
type_ = param.typeName().getText()
params.append(Param(type=types.Placeholder(type_), name=var_name))
body = self.visit(ctx.functionBody())
return Function(
# Note: The function name is the internally generated name for this module
name=function_name,
is_exported=ctx.EXPORT() is not None,
params=params,
type=types.Placeholder(text=return_type),
body=body,
)
def visitTrait(self, ctx: BugParser.TraitContext):
name = ctx.traitName().getText()
functions = []
for function in ctx.traitFunctionDef():
functions.append(self.visit(function))
return Trait(
name=name,
is_exported=False,
functions=functions,
type=types.Placeholder(text=name),
)
def test_empty_data():
source = """
import System::Output;
export data Empty()
export function Main(): void {
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[Import(value="System::Output")],
traits=[],
data_defs=[
DataDef(
name="Empty",
implements=[],
type=types.Placeholder(text="Empty"),
is_exported=True,
params=[],
functions=[],
)
],
functions=[
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[],
)
],
),
)
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_trait_with_no_functions():
source = """
trait List {
}
"""
result = get_ast(source)
assert_equal_programs(
result,
Program(
imports=[],
traits=[
Trait(
name="List",
is_exported=False,
type=types.Placeholder("List"),
functions=[],
)
],
data_defs=[],
functions=[],
),
)
def test_data_vector_with_complex_function():
source = """
import System::Output;
export data Vector(x: i32, y: i32) {
function add(self: Vector, other: Vector): Vector {
Vector(x = self.x + other.x, y = self.y + other.y);
}
}
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="add",
is_exported=False,
params=[
Param(name="self",
type=types.Placeholder(text="Vector")),
Param(name="other",
type=types.Placeholder(text="Vector")),
],
type=types.Placeholder(text="Vector"),
body=[
FunctionCall(
name="Vector",
arguments=[
Argument(
name="x",
value=BinaryOperation(
operator=BinaryOperator.ADD,
left=MemberAccess(
value=Variable("self"),
member="x"),
right=MemberAccess(
value=Variable("other"),
member="x"),
),
),
Argument(
name="y",
value=BinaryOperation(
operator=BinaryOperator.ADD,
left=MemberAccess(
value=Variable("self"),
member="y"),
right=MemberAccess(
value=Variable("other"),
member="y"),
),
),
],
)
],
)
],
)
],
functions=[
Function(
name="Main",
is_exported=True,
params=[],
type=types.Placeholder(text="void"),
body=[],
)
],
),
)
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_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 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_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_simple_predicates():
source = """
function isGreaterThan(left: i32, right: i32): boolean {
left > right;
}
function isLessThan(left: i32, right: i32): boolean {
left < right;
}
function and(left: boolean, right: boolean): boolean {
left && right;
}
function or(left: boolean, right: boolean): boolean {
left || right;
}
"""
result = get_ast(source)
assert_equal_programs(
result,
compiler.ast.Program(
imports=[],
traits=[],
data_defs=[],
functions=[
compiler.ast.Function(
name="isGreaterThan",
is_exported=False,
params=[
compiler.ast.Param(name="left",
type=types.Placeholder(text="i32")),
compiler.ast.Param(name="right",
type=types.Placeholder(text="i32")),
],
type=types.Placeholder(text="boolean"),
body=[
compiler.ast.BinaryOperation(
operator=compiler.ast.BinaryOperator.GREATER_THAN,
left=compiler.ast.Variable(name="left"),
right=compiler.ast.Variable(name="right"),
)
],
),
compiler.ast.Function(
name="isLessThan",
is_exported=False,
params=[
compiler.ast.Param(name="left",
type=types.Placeholder(text="i32")),
compiler.ast.Param(name="right",
type=types.Placeholder(text="i32")),
],
type=types.Placeholder(text="boolean"),
body=[
compiler.ast.BinaryOperation(
operator=compiler.ast.BinaryOperator.LESS_THAN,
left=compiler.ast.Variable(name="left"),
right=compiler.ast.Variable(name="right"),
)
],
),
compiler.ast.Function(
name="and",
is_exported=False,
params=[
compiler.ast.Param(
name="left",
type=types.Placeholder(text="boolean")),
compiler.ast.Param(
name="right",
type=types.Placeholder(text="boolean")),
],
type=types.Placeholder(text="boolean"),
body=[
compiler.ast.BinaryOperation(
operator=compiler.ast.BinaryOperator.AND,
left=compiler.ast.Variable(name="left"),
right=compiler.ast.Variable(name="right"),
)
],
),
compiler.ast.Function(
name="or",
is_exported=False,
params=[
compiler.ast.Param(
name="left",
type=types.Placeholder(text="boolean")),
compiler.ast.Param(
name="right",
type=types.Placeholder(text="boolean")),
],
type=types.Placeholder(text="boolean"),
body=[
compiler.ast.BinaryOperation(
operator=compiler.ast.BinaryOperator.OR,
left=compiler.ast.Variable(name="left"),
right=compiler.ast.Variable(name="right"),
)
],
),
],
),
)
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)))
],
),
)
],
)
],
),
],
),
)
