probablycorey/ReefVM
1
0
Fork 0
forked from defunkt/ReefVM
Code Pull Requests Activity

simplify

Browse Source
This commit is contained in:
Chris Wanstrath 2025-10-14 12:18:55 -07:00
parent df9af925d3
commit 173fd28d6f
1 changed files with 27 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
tests/functions.test.ts
View File

@ -1,13 +1,13 @@
import { test, expect } from "bun:test"
import { toBytecode } from "#bytecode"
import { VM } from "#vm"
import { toValue, run } from "#reef"
test("MAKE_FUNCTION - creates function with captured scope", async () => {
const bytecode = toBytecode(`
MAKE_FUNCTION () #999
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result.type).toBe('function')
if (result.type === 'function') {
expect(result.body).toBe(999)
@ -26,7 +26,7 @@ test("CALL and RETURN - basic function call", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({ type: 'number', value: 42 })
})
@ -42,7 +42,7 @@ test("CALL and RETURN - function with one parameter", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({ type: 'number', value: 100 })
})
@ -61,7 +61,7 @@ test("CALL and RETURN - function with two parameters", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({ type: 'number', value: 30 })
})
@ -79,7 +79,7 @@ test("CALL - variadic function with no fixed params", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({
type: 'array',
value: [
@ -104,7 +104,7 @@ test("CALL - variadic function with one fixed param", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// x should be 10, rest should be [20, 30]
expect(result).toEqual({
type: 'array',
@ -130,7 +130,7 @@ test("CALL - variadic function with two fixed params", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// a=1, b=2, rest=[3, 4]
expect(result).toEqual({
type: 'array',
@ -153,7 +153,7 @@ test("CALL - variadic function with no extra args", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// rest should be empty array
expect(result).toEqual({ type: 'array', value: [] })
})
@ -169,7 +169,7 @@ test("CALL - variadic function with defaults on fixed params", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// x should use default value 5
expect(result).toEqual({ type: 'number', value: 5 })
})
@ -188,7 +188,7 @@ test("TAIL_CALL - variadic function", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// Should return the rest array [2, 3]
expect(result).toEqual({
type: 'array',
@ -214,7 +214,7 @@ test("CALL - named args function with no fixed params", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result.type).toBe('dict')
if (result.type === 'dict') {
expect(result.value.get('name')).toEqual({ type: 'string', value: 'Bob' })
@ -236,7 +236,7 @@ test("CALL - named args function with one fixed param", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result.type).toBe('dict')
if (result.type === 'dict') {
expect(result.value.get('name')).toEqual({ type: 'string', value: 'Alice' })
@ -258,7 +258,7 @@ test("CALL - named args with matching param name should bind to param not named"
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// name should be bound as regular param, not collected in named
expect(result).toEqual({ type: 'string', value: 'Bob' })
})
@ -278,7 +278,7 @@ test("CALL - named args that match param names should not be in named", async ()
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result.type).toBe('dict')
if (result.type === 'dict') {
// Only city should be in named, name should be bound to param
@ -304,7 +304,7 @@ test("CALL - mixed variadic and named args", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// rest should have [2, 3]
expect(result).toEqual({
type: 'array',
@ -331,7 +331,7 @@ test("CALL - mixed variadic and named args, check named", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result.type).toBe('dict')
if (result.type === 'dict') {
expect(result.value.get('name')).toEqual({ type: 'string', value: 'Bob' })
@ -350,7 +350,7 @@ test("CALL - named args with no extra named args", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// named should be empty dict
expect(result.type).toBe('dict')
if (result.type === 'dict') {
@ -371,7 +371,7 @@ test("CALL - named args with defaults on fixed params", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// x should use default value 5
expect(result).toEqual({ type: 'number', value: 5 })
})
@ -387,7 +387,7 @@ test("TRY_CALL - calls function if found", async () => {
["RETURN"]
])
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({ type: 'number', value: 42 })
})
@ -399,7 +399,7 @@ test("TRY_CALL - pushes value if variable exists but is not a function", async (
["HALT"]
])
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({ type: 'number', value: 99 })
})
@ -409,7 +409,7 @@ test("TRY_CALL - pushes string if variable not found", async () => {
["HALT"]
])
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({ type: 'string', value: 'unknownVar' })
})
@ -423,7 +423,7 @@ test("TRY_CALL - handles arrays", async () => {
["HALT"]
])
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result.type).toBe('array')
if (result.type === 'array') {
expect(result.value).toEqual([
@ -443,7 +443,7 @@ test("TRY_CALL - handles dicts", async () => {
["HALT"]
])
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result.type).toBe('dict')
if (result.type === 'dict') {
expect(result.value.get('key')).toEqual({ type: 'string', value: 'value' })
@ -458,7 +458,7 @@ test("TRY_CALL - handles null values", async () => {
["HALT"]
])
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({ type: 'null', value: null })
})
@ -477,7 +477,7 @@ test("TRY_CALL - function can access its parameters", async () => {
["RETURN"]
])
const result = await new VM(bytecode).run()
const result = await run(bytecode)
// Function is called with 0 args, so x inside function should be null
// Then we add 5 to null (which coerces to 0)
expect(result).toEqual({ type: 'number', value: 5 })
@ -493,6 +493,6 @@ test("TRY_CALL - with string format", async () => {
RETURN
`)
const result = await new VM(bytecode).run()
const result = await run(bytecode)
expect(result).toEqual({ type: 'number', value: 100 })
})