def test_parallelism():
one_plus_one_ast = lex_and_parse("""(|
add = (|| 1 + 1)
|) add""")
two_plus_two_ast = lex_and_parse("""(|
add = (|| 2 + 2)
|) add""")
one_plus_one_ctx = one_plus_one_ast[0].compile(CodeContext())
two_plus_two_ctx = two_plus_two_ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives())
one_plus_one_process = interpreter.add_process(one_plus_one_ctx)
two_plus_two_process = interpreter.add_process(two_plus_two_ctx)
assert not one_plus_one_process.finished
assert not one_plus_one_process.result
assert not two_plus_two_process.finished
assert not two_plus_two_process.result
interpreter.interpret()
assert one_plus_one_process.finished
assert one_plus_one_process.result == PrimitiveIntObject(2)
assert two_plus_two_process.finished
assert two_plus_two_process.result == PrimitiveIntObject(4)
def test_loop(self):
source = """
(|
benchmark = (| i <- 0. |
[i < 1000] whileTrue: [
i: i + 1.
].
).
|) benchmark.
"""
universe = Object()
universe.meta_add_slot("primitives", get_primitives())
interpreter = Interpreter(universe)
run_stdlib(interpreter, stdlib_source)
ast = lex_and_parse_as_root(source)
if not ast:
return None, interpreter
code = ast.compile(CodeContext())
interpreter.add_process(code.finalize())
def f():
interpreter.interpret()
return 0
#res = f()
#assert res == 0
res = self.meta_interp(f, [],
listops=True,
listcomp=True,
backendopt=True)
assert res == 0
def test_int_and_float():
ast = lex_and_parse("""(|
add = (||
1 + 1.5
)
|) add""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
assert interpreter.process.result == PrimitiveFloatObject(2.5)
def test_setting_of_error_handler_works():
ast = lex_and_parse("""(|
test = (||
primitives interpreter setErrorHandler: [:obj. :stack |
obj printString.
]
)
|) test""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
def test_interpreter():
ast = lex_and_parse("""(|
a <- 1.
add: b = (||
(a + 1) + b
)
|) add: 2""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
assert interpreter.process.result == PrimitiveIntObject(4)
def test_resend_keyword():
ast = lex_and_parse("""(|
p* = (| xex: a = (|| 1). |).
q* = (| xex: a = (|| a). |).
fetchXex = (|| q.xex: 2)
|) fetchXex""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
assert interpreter.process.result == PrimitiveIntObject(2)
def test_assignment_primitive():
ast = lex_and_parse("""(|
a <- 0.
incA = (||
a: 1.
^a
)
|) incA""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
assert interpreter.process.result == PrimitiveIntObject(1)
def test_unhandled_error():
ast = lex_and_parse("""(|
test = (||
primitives interpreter error: 'Test'
)
|) test""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
assert interpreter.process.finished
assert interpreter.process.finished_with_error
assert interpreter.process.result == PrimitiveStrObject("Test")
def test_running_self_unittest_file():
universe = Object()
universe.meta_add_slot("primitives", get_primitives())
dirname = os.path.dirname(__file__)
source_file_path = os.path.join(dirname, "..", "scripts", "unittest.self")
with open(source_file_path) as source_file:
ast = lex_and_parse(source_file.read())
interpreter = Interpreter(universe)
for item in ast:
process = interpreter.add_process(item.compile(CodeContext()))
interpreter.interpret()
assert process.finished
assert not process.finished_with_error
def test_block_with_argument():
ast = lex_and_parse("""(|
a <- 0.
giveOneToBlock: blk = (||
blk with: 1
).
shouldBeThree = (| tmp = 1. block. |
block: [| :x | a + tmp + x].
a: 1.
giveOneToBlock: block.
)
|) shouldBeThree""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
assert interpreter.process.result == PrimitiveIntObject(3)
def test_block():
ast = lex_and_parse("""(|
a <- 0.
addTenToBlk: blk = (||
10 + (blk value).
).
add = (| tmp = 3. tmp_block. |
tmp_block: [a + tmp].
a: 1. # block should reflect current scopes
2 + addTenToBlk: tmp_block.
)
|) add""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
assert interpreter.process.result == PrimitiveIntObject(16)
def test_set_error_handler_and_handle_error():
ast = lex_and_parse("""(|
raiseError = (|| primitives interpreter error: 'Test'.).
test = (||
primitives interpreter setErrorHandler: [:msg. :err |
primitives interpreter restoreProcess: err With: 1.
].
^ 1 + raiseError.
)
|) test""")
context = ast[0].compile(CodeContext())
interpreter = Interpreter(universe=get_primitives(), code_context=context)
interpreter.interpret()
assert interpreter.process.finished
assert not interpreter.process.finished_with_error
result = interpreter.process.result
assert result == PrimitiveIntObject(2)
def virtual_machine(source, stdlib_source=""):
universe = Object()
universe.meta_add_slot("primitives", get_primitives())
interpreter = Interpreter(universe)
if stdlib_source:
if not run_stdlib(interpreter, stdlib_source):
return None, interpreter
ast = lex_and_parse_as_root(source)
if not ast:
return None, interpreter
code = ast.compile(CodeContext())
process = interpreter.add_process(code.finalize())
interpreter.interpret()
return process, interpreter
