probablycorey/shrimp
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Merge branch 'main' into less-parsing

Browse Source
This commit is contained in:
probablycorey 2025-11-07 15:28:12 +00:00
commit c325bca611
3 changed files with 171 additions and 190 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/compiler/tests/pipe.test.ts
View File

@ -89,7 +89,7 @@ describe('pipe expressions', () => {
test('pipe with prelude function (echo)', () => { test('pipe with prelude function (echo)', () => {
expect(` expect(`
get-msg = do: 'hello' end get-msg = do: 'hello' end
get-msg | echo get-msg | length
`).toEvaluateTo(null) `).toEvaluateTo(5)
}) })
}) })

2
src/compiler/tests/ribbit.test.ts
View File

@ -83,7 +83,7 @@ end
test('custom tags', () => { test('custom tags', () => {
expect(` expect(`
list = tag ul class=list list = tag ul class='list'
ribbit: ribbit:
list: list:
li border-bottom='1px solid black' one li border-bottom='1px solid black' one

355
src/prelude/tests/prelude.test.ts
View File

@ -3,185 +3,185 @@ import { globals } from '#prelude'
describe('string operations', () => { describe('string operations', () => {
test('to-upper converts to uppercase', async () => { test('to-upper converts to uppercase', async () => {
await expect(`str.to-upper 'hello'`).toEvaluateTo('HELLO', globals) await expect(`str.to-upper 'hello'`).toEvaluateTo('HELLO')
await expect(`str.to-upper 'Hello World!'`).toEvaluateTo('HELLO WORLD!', globals) await expect(`str.to-upper 'Hello World!'`).toEvaluateTo('HELLO WORLD!')
}) })
test('to-lower converts to lowercase', async () => { test('to-lower converts to lowercase', async () => {
await expect(`str.to-lower 'HELLO'`).toEvaluateTo('hello', globals) await expect(`str.to-lower 'HELLO'`).toEvaluateTo('hello')
await expect(`str.to-lower 'Hello World!'`).toEvaluateTo('hello world!', globals) await expect(`str.to-lower 'Hello World!'`).toEvaluateTo('hello world!')
}) })
test('trim removes whitespace', async () => { test('trim removes whitespace', async () => {
await expect(`str.trim ' hello '`).toEvaluateTo('hello', globals) await expect(`str.trim ' hello '`).toEvaluateTo('hello')
await expect(`str.trim '\\n\\thello\\t\\n'`).toEvaluateTo('hello', globals) await expect(`str.trim '\\n\\thello\\t\\n'`).toEvaluateTo('hello')
}) })
test('split divides string by separator', async () => { test('split divides string by separator', async () => {
await expect(`str.split 'a,b,c' ','`).toEvaluateTo(['a', 'b', 'c'], globals) await expect(`str.split 'a,b,c' ','`).toEvaluateTo(['a', 'b', 'c'])
await expect(`str.split 'hello' ''`).toEvaluateTo(['h', 'e', 'l', 'l', 'o'], globals) await expect(`str.split 'hello' ''`).toEvaluateTo(['h', 'e', 'l', 'l', 'o'])
}) })
test('split with comma separator', async () => { test('split with comma separator', async () => {
await expect(`str.split 'a,b,c' ','`).toEvaluateTo(['a', 'b', 'c'], globals) await expect(`str.split 'a,b,c' ','`).toEvaluateTo(['a', 'b', 'c'])
}) })
test('join combines array elements', async () => { test('join combines array elements', async () => {
await expect(`str.join ['a' 'b' 'c'] '-'`).toEvaluateTo('a-b-c', globals) await expect(`str.join ['a' 'b' 'c'] '-'`).toEvaluateTo('a-b-c')
await expect(`str.join ['hello' 'world'] ' '`).toEvaluateTo('hello world', globals) await expect(`str.join ['hello' 'world'] ' '`).toEvaluateTo('hello world')
}) })
test('join with comma separator', async () => { test('join with comma separator', async () => {
await expect(`str.join ['a' 'b' 'c'] ','`).toEvaluateTo('a,b,c', globals) await expect(`str.join ['a' 'b' 'c'] ','`).toEvaluateTo('a,b,c')
}) })
test('starts-with? checks string prefix', async () => { test('starts-with? checks string prefix', async () => {
await expect(`str.starts-with? 'hello' 'hel'`).toEvaluateTo(true, globals) await expect(`str.starts-with? 'hello' 'hel'`).toEvaluateTo(true)
await expect(`str.starts-with? 'hello' 'bye'`).toEvaluateTo(false, globals) await expect(`str.starts-with? 'hello' 'bye'`).toEvaluateTo(false)
}) })
test('ends-with? checks string suffix', async () => { test('ends-with? checks string suffix', async () => {
await expect(`str.ends-with? 'hello' 'lo'`).toEvaluateTo(true, globals) await expect(`str.ends-with? 'hello' 'lo'`).toEvaluateTo(true)
await expect(`str.ends-with? 'hello' 'he'`).toEvaluateTo(false, globals) await expect(`str.ends-with? 'hello' 'he'`).toEvaluateTo(false)
}) })
test('contains? checks for substring', async () => { test('contains? checks for substring', async () => {
await expect(`str.contains? 'hello world' 'o w'`).toEvaluateTo(true, globals) await expect(`str.contains? 'hello world' 'o w'`).toEvaluateTo(true)
await expect(`str.contains? 'hello' 'bye'`).toEvaluateTo(false, globals) await expect(`str.contains? 'hello' 'bye'`).toEvaluateTo(false)
}) })
test('empty? checks if string is empty', async () => { test('empty? checks if string is empty', async () => {
await expect(`str.empty? ''`).toEvaluateTo(true, globals) await expect(`str.empty? ''`).toEvaluateTo(true)
await expect(`str.empty? 'hello'`).toEvaluateTo(false, globals) await expect(`str.empty? 'hello'`).toEvaluateTo(false)
}) })
test('replace replaces first occurrence', async () => { test('replace replaces first occurrence', async () => {
await expect(`str.replace 'hello hello' 'hello' 'hi'`).toEvaluateTo('hi hello', globals) await expect(`str.replace 'hello hello' 'hello' 'hi'`).toEvaluateTo('hi hello')
}) })
test('replace-all replaces all occurrences', async () => { test('replace-all replaces all occurrences', async () => {
await expect(`str.replace-all 'hello hello' 'hello' 'hi'`).toEvaluateTo('hi hi', globals) await expect(`str.replace-all 'hello hello' 'hello' 'hi'`).toEvaluateTo('hi hi')
}) })
test('slice extracts substring', async () => { test('slice extracts substring', async () => {
await expect(`str.slice 'hello' 1 3`).toEvaluateTo('el', globals) await expect(`str.slice 'hello' 1 3`).toEvaluateTo('el')
await expect(`str.slice 'hello' 2 null`).toEvaluateTo('llo', globals) await expect(`str.slice 'hello' 2 null`).toEvaluateTo('llo')
}) })
test('repeat repeats string', async () => { test('repeat repeats string', async () => {
await expect(`str.repeat 'ha' 3`).toEvaluateTo('hahaha', globals) await expect(`str.repeat 'ha' 3`).toEvaluateTo('hahaha')
}) })
test('pad-start pads beginning', async () => { test('pad-start pads beginning', async () => {
await expect(`str.pad-start '5' 3 '0'`).toEvaluateTo('005', globals) await expect(`str.pad-start '5' 3 '0'`).toEvaluateTo('005')
}) })
test('pad-end pads end', async () => { test('pad-end pads end', async () => {
await expect(`str.pad-end '5' 3 '0'`).toEvaluateTo('500', globals) await expect(`str.pad-end '5' 3 '0'`).toEvaluateTo('500')
}) })
test('lines splits by newlines', async () => { test('lines splits by newlines', async () => {
await expect(`str.lines 'a\\nb\\nc'`).toEvaluateTo(['a', 'b', 'c'], globals) await expect(`str.lines 'a\\nb\\nc'`).toEvaluateTo(['a', 'b', 'c'])
}) })
test('chars splits into characters', async () => { test('chars splits into characters', async () => {
await expect(`str.chars 'abc'`).toEvaluateTo(['a', 'b', 'c'], globals) await expect(`str.chars 'abc'`).toEvaluateTo(['a', 'b', 'c'])
}) })
test('index-of finds substring position', async () => { test('index-of finds substring position', async () => {
await expect(`str.index-of 'hello world' 'world'`).toEvaluateTo(6, globals) await expect(`str.index-of 'hello world' 'world'`).toEvaluateTo(6)
await expect(`str.index-of 'hello' 'bye'`).toEvaluateTo(-1, globals) await expect(`str.index-of 'hello' 'bye'`).toEvaluateTo(-1)
}) })
test('last-index-of finds last occurrence', async () => { test('last-index-of finds last occurrence', async () => {
await expect(`str.last-index-of 'hello hello' 'hello'`).toEvaluateTo(6, globals) await expect(`str.last-index-of 'hello hello' 'hello'`).toEvaluateTo(6)
}) })
}) })
describe('boolean logic', () => { describe('boolean logic', () => {
test('not negates value', async () => { test('not negates value', async () => {
await expect(`not true`).toEvaluateTo(false, globals) await expect(`not true`).toEvaluateTo(false)
await expect(`not false`).toEvaluateTo(true, globals) await expect(`not false`).toEvaluateTo(true)
await expect(`not 42`).toEvaluateTo(false, globals) await expect(`not 42`).toEvaluateTo(false)
await expect(`not null`).toEvaluateTo(true, globals) await expect(`not null`).toEvaluateTo(true)
}) })
}) })
describe('utilities', () => { describe('utilities', () => {
test('inc increments by 1', async () => { test('inc increments by 1', async () => {
await expect(`inc 5`).toEvaluateTo(6, globals) await expect(`inc 5`).toEvaluateTo(6)
await expect(`inc -1`).toEvaluateTo(0, globals) await expect(`inc -1`).toEvaluateTo(0)
}) })
test('dec decrements by 1', async () => { test('dec decrements by 1', async () => {
await expect(`dec 5`).toEvaluateTo(4, globals) await expect(`dec 5`).toEvaluateTo(4)
await expect(`dec 0`).toEvaluateTo(-1, globals) await expect(`dec 0`).toEvaluateTo(-1)
}) })
test('identity returns value as-is', async () => { test('identity returns value as-is', async () => {
await expect(`identity 42`).toEvaluateTo(42, globals) await expect(`identity 42`).toEvaluateTo(42)
await expect(`identity 'hello'`).toEvaluateTo('hello', globals) await expect(`identity 'hello'`).toEvaluateTo('hello')
}) })
}) })
describe('collections', () => { describe('collections', () => {
test('length', async () => { test('length', async () => {
await expect(`length 'hello'`).toEvaluateTo(5, globals) await expect(`length 'hello'`).toEvaluateTo(5)
await expect(`length [1 2 3]`).toEvaluateTo(3, globals) await expect(`length [1 2 3]`).toEvaluateTo(3)
await expect(`length [a=1 b=2]`).toEvaluateTo(2, globals) await expect(`length [a=1 b=2]`).toEvaluateTo(2)
}) })
test('length throws on invalid types', async () => { test('length throws on invalid types', async () => {
await expect(`try: length 42 catch e: 'error' end`).toEvaluateTo('error', globals) await expect(`try: length 42 catch e: 'error' end`).toEvaluateTo('error')
await expect(`try: length true catch e: 'error' end`).toEvaluateTo('error', globals) await expect(`try: length true catch e: 'error' end`).toEvaluateTo('error')
await expect(`try: length null catch e: 'error' end`).toEvaluateTo('error', globals) await expect(`try: length null catch e: 'error' end`).toEvaluateTo('error')
}) })
test('literal array creates array from arguments', async () => { test('literal array creates array from arguments', async () => {
await expect(`[ 1 2 3 ]`).toEvaluateTo([1, 2, 3], globals) await expect(`[ 1 2 3 ]`).toEvaluateTo([1, 2, 3])
await expect(`['a' 'b']`).toEvaluateTo(['a', 'b'], globals) await expect(`['a' 'b']`).toEvaluateTo(['a', 'b'])
await expect(`[]`).toEvaluateTo([], globals) await expect(`[]`).toEvaluateTo([])
}) })
test('literal dict creates object from named arguments', async () => { test('literal dict creates object from named arguments', async () => {
await expect(`[ a=1 b=2 ]`).toEvaluateTo({ a: 1, b: 2 }, globals) await expect(`[ a=1 b=2 ]`).toEvaluateTo({ a: 1, b: 2 })
await expect(`[=]`).toEvaluateTo({}, globals) await expect(`[=]`).toEvaluateTo({})
}) })
test('at retrieves element at index', async () => { test('at retrieves element at index', async () => {
await expect(`at [10 20 30] 0`).toEvaluateTo(10, globals) await expect(`at [10 20 30] 0`).toEvaluateTo(10)
await expect(`at [10 20 30] 2`).toEvaluateTo(30, globals) await expect(`at [10 20 30] 2`).toEvaluateTo(30)
}) })
test('at retrieves property from object', async () => { test('at retrieves property from object', async () => {
await expect(`at [name='test'] 'name'`).toEvaluateTo('test', globals) await expect(`at [name='test'] 'name'`).toEvaluateTo('test')
}) })
test('slice extracts array subset', async () => { test('slice extracts array subset', async () => {
await expect(`list.slice [1 2 3 4 5] 1 3`).toEvaluateTo([2, 3], globals) await expect(`list.slice [1 2 3 4 5] 1 3`).toEvaluateTo([2, 3])
await expect(`list.slice [1 2 3 4 5] 2 5`).toEvaluateTo([3, 4, 5], globals) await expect(`list.slice [1 2 3 4 5] 2 5`).toEvaluateTo([3, 4, 5])
}) })
test('range creates number sequence', async () => { test('range creates number sequence', async () => {
await expect(`range 0 5`).toEvaluateTo([0, 1, 2, 3, 4, 5], globals) await expect(`range 0 5`).toEvaluateTo([0, 1, 2, 3, 4, 5])
await expect(`range 3 6`).toEvaluateTo([3, 4, 5, 6], globals) await expect(`range 3 6`).toEvaluateTo([3, 4, 5, 6])
}) })
test('range with single argument starts from 0', async () => { test('range with single argument starts from 0', async () => {
await expect(`range 3 null`).toEvaluateTo([0, 1, 2, 3], globals) await expect(`range 3 null`).toEvaluateTo([0, 1, 2, 3])
await expect(`range 0 null`).toEvaluateTo([0], globals) await expect(`range 0 null`).toEvaluateTo([0])
}) })
test('empty? checks if list, dict, string is empty', async () => { test('empty? checks if list, dict, string is empty', async () => {
await expect(`empty? []`).toEvaluateTo(true, globals) await expect(`empty? []`).toEvaluateTo(true)
await expect(`empty? [1]`).toEvaluateTo(false, globals) await expect(`empty? [1]`).toEvaluateTo(false)
await expect(`empty? [=]`).toEvaluateTo(true, globals) await expect(`empty? [=]`).toEvaluateTo(true)
await expect(`empty? [a=true]`).toEvaluateTo(false, globals) await expect(`empty? [a=true]`).toEvaluateTo(false)
await expect(`empty? ''`).toEvaluateTo(true, globals) await expect(`empty? ''`).toEvaluateTo(true)
await expect(`empty? 'cat'`).toEvaluateTo(false, globals) await expect(`empty? 'cat'`).toEvaluateTo(false)
await expect(`empty? meow`).toEvaluateTo(false, globals) await expect(`empty? meow`).toEvaluateTo(false)
}) })
test('list.filter keeps matching elements', async () => { test('list.filter keeps matching elements', async () => {
@ -190,7 +190,7 @@ describe('collections', () => {
x == 3 or x == 4 or x == 5 x == 3 or x == 4 or x == 5
end end
list.filter [1 2 3 4 5] is-positive list.filter [1 2 3 4 5] is-positive
`).toEvaluateTo([3, 4, 5], globals) `).toEvaluateTo([3, 4, 5])
}) })
test('list.reduce accumulates values', async () => { test('list.reduce accumulates values', async () => {
@ -199,7 +199,7 @@ describe('collections', () => {
acc + x acc + x
end end
list.reduce [1 2 3 4] add 0 list.reduce [1 2 3 4] add 0
`).toEvaluateTo(10, globals) `).toEvaluateTo(10)
}) })
test('list.find returns first match', async () => { test('list.find returns first match', async () => {
@ -208,139 +208,139 @@ describe('collections', () => {
x == 4 x == 4
end end
list.find [1 2 4 5] is-four list.find [1 2 4 5] is-four
`).toEvaluateTo(4, globals) `).toEvaluateTo(4)
}) })
test('list.find returns null if no match', async () => { test('list.find returns null if no match', async () => {
await expect(` await expect(`
is-ten = do x: x == 10 end is-ten = do x: x == 10 end
list.find [1 2 3] is-ten list.find [1 2 3] is-ten
`).toEvaluateTo(null, globals) `).toEvaluateTo(null)
}) })
test('list.empty? checks if list is empty', async () => { test('list.empty? checks if list is empty', async () => {
await expect(`list.empty? []`).toEvaluateTo(true, globals) await expect(`list.empty? []`).toEvaluateTo(true)
await expect(`list.empty? [1]`).toEvaluateTo(false, globals) await expect(`list.empty? [1]`).toEvaluateTo(false)
}) })
test('list.contains? checks for element', async () => { test('list.contains? checks for element', async () => {
await expect(`list.contains? [1 2 3] 2`).toEvaluateTo(true, globals) await expect(`list.contains? [1 2 3] 2`).toEvaluateTo(true)
await expect(`list.contains? [1 2 3] 5`).toEvaluateTo(false, globals) await expect(`list.contains? [1 2 3] 5`).toEvaluateTo(false)
}) })
test('list.reverse reverses array', async () => { test('list.reverse reverses array', async () => {
await expect(`list.reverse [1 2 3]`).toEvaluateTo([3, 2, 1], globals) await expect(`list.reverse [1 2 3]`).toEvaluateTo([3, 2, 1])
}) })
test('list.concat combines arrays', async () => { test('list.concat combines arrays', async () => {
await expect(`list.concat [1 2] [3 4]`).toEvaluateTo([1, 2, 3, 4], globals) await expect(`list.concat [1 2] [3 4]`).toEvaluateTo([1, 2, 3, 4])
}) })
test('list.flatten flattens nested arrays', async () => { test('list.flatten flattens nested arrays', async () => {
await expect(`list.flatten [[1 2] [3 4]] 1`).toEvaluateTo([1, 2, 3, 4], globals) await expect(`list.flatten [[1 2] [3 4]] 1`).toEvaluateTo([1, 2, 3, 4])
}) })
test('list.unique removes duplicates', async () => { test('list.unique removes duplicates', async () => {
await expect(`list.unique [1 2 2 3 1]`).toEvaluateTo([1, 2, 3], globals) await expect(`list.unique [1 2 2 3 1]`).toEvaluateTo([1, 2, 3])
}) })
test('list.zip combines two arrays', async () => { test('list.zip combines two arrays', async () => {
await expect(`list.zip [1 2] [3 4]`).toEvaluateTo([[1, 3], [2, 4]], globals) await expect(`list.zip [1 2] [3 4]`).toEvaluateTo([[1, 3], [2, 4]])
}) })
test('list.first returns first element', async () => { test('list.first returns first element', async () => {
await expect(`list.first [1 2 3]`).toEvaluateTo(1, globals) await expect(`list.first [1 2 3]`).toEvaluateTo(1)
await expect(`list.first []`).toEvaluateTo(null, globals) await expect(`list.first []`).toEvaluateTo(null)
}) })
test('list.last returns last element', async () => { test('list.last returns last element', async () => {
await expect(`list.last [1 2 3]`).toEvaluateTo(3, globals) await expect(`list.last [1 2 3]`).toEvaluateTo(3)
await expect(`list.last []`).toEvaluateTo(null, globals) await expect(`list.last []`).toEvaluateTo(null)
}) })
test('list.rest returns all but first', async () => { test('list.rest returns all but first', async () => {
await expect(`list.rest [1 2 3]`).toEvaluateTo([2, 3], globals) await expect(`list.rest [1 2 3]`).toEvaluateTo([2, 3])
}) })
test('list.take returns first n elements', async () => { test('list.take returns first n elements', async () => {
await expect(`list.take [1 2 3 4 5] 3`).toEvaluateTo([1, 2, 3], globals) await expect(`list.take [1 2 3 4 5] 3`).toEvaluateTo([1, 2, 3])
}) })
test('list.drop skips first n elements', async () => { test('list.drop skips first n elements', async () => {
await expect(`list.drop [1 2 3 4 5] 2`).toEvaluateTo([3, 4, 5], globals) await expect(`list.drop [1 2 3 4 5] 2`).toEvaluateTo([3, 4, 5])
}) })
test('list.append adds to end', async () => { test('list.append adds to end', async () => {
await expect(`list.append [1 2] 3`).toEvaluateTo([1, 2, 3], globals) await expect(`list.append [1 2] 3`).toEvaluateTo([1, 2, 3])
}) })
test('list.prepend adds to start', async () => { test('list.prepend adds to start', async () => {
await expect(`list.prepend [2 3] 1`).toEvaluateTo([1, 2, 3], globals) await expect(`list.prepend [2 3] 1`).toEvaluateTo([1, 2, 3])
}) })
test('list.index-of finds element index', async () => { test('list.index-of finds element index', async () => {
await expect(`list.index-of [1 2 3] 2`).toEvaluateTo(1, globals) await expect(`list.index-of [1 2 3] 2`).toEvaluateTo(1)
await expect(`list.index-of [1 2 3] 5`).toEvaluateTo(-1, globals) await expect(`list.index-of [1 2 3] 5`).toEvaluateTo(-1)
}) })
test('list.push adds to end and mutates array', async () => { test('list.push adds to end and mutates array', async () => {
await expect(`arr = [1 2]; list.push arr 3; arr`).toEvaluateTo([1, 2, 3], globals) await expect(`arr = [1 2]; list.push arr 3; arr`).toEvaluateTo([1, 2, 3])
}) })
test('list.push returns the size of the array', async () => { test('list.push returns the size of the array', async () => {
await expect(`arr = [1 2]; arr | list.push 3`).toEvaluateTo(3, globals) await expect(`arr = [1 2]; arr | list.push 3`).toEvaluateTo(3)
}) })
test('list.pop removes from end and mutates array', async () => { test('list.pop removes from end and mutates array', async () => {
await expect(`arr = [1 2 3]; list.pop arr; arr`).toEvaluateTo([1, 2], globals) await expect(`arr = [1 2 3]; list.pop arr; arr`).toEvaluateTo([1, 2])
}) })
test('list.pop returns removed element', async () => { test('list.pop returns removed element', async () => {
await expect(`list.pop [1 2 3]`).toEvaluateTo(3, globals) await expect(`list.pop [1 2 3]`).toEvaluateTo(3)
}) })
test('list.pop returns null for empty array', async () => { test('list.pop returns null for empty array', async () => {
await expect(`list.pop []`).toEvaluateTo(null, globals) await expect(`list.pop []`).toEvaluateTo(null)
}) })
test('list.shift removes from start and mutates array', async () => { test('list.shift removes from start and mutates array', async () => {
await expect(`arr = [1 2 3]; list.shift arr; arr`).toEvaluateTo([2, 3], globals) await expect(`arr = [1 2 3]; list.shift arr; arr`).toEvaluateTo([2, 3])
}) })
test('list.shift returns removed element', async () => { test('list.shift returns removed element', async () => {
await expect(`list.shift [1 2 3]`).toEvaluateTo(1, globals) await expect(`list.shift [1 2 3]`).toEvaluateTo(1)
}) })
test('list.shift returns null for empty array', async () => { test('list.shift returns null for empty array', async () => {
await expect(`list.shift []`).toEvaluateTo(null, globals) await expect(`list.shift []`).toEvaluateTo(null)
}) })
test('list.unshift adds to start and mutates array', async () => { test('list.unshift adds to start and mutates array', async () => {
await expect(`arr = [2 3]; list.unshift arr 1; arr`).toEvaluateTo([1, 2, 3], globals) await expect(`arr = [2 3]; list.unshift arr 1; arr`).toEvaluateTo([1, 2, 3])
}) })
test('list.unshift returns the length of the array', async () => { test('list.unshift returns the length of the array', async () => {
await expect(`arr = [2 3]; arr | list.unshift 1`).toEvaluateTo(3, globals) await expect(`arr = [2 3]; arr | list.unshift 1`).toEvaluateTo(3)
}) })
test('list.splice removes elements and mutates array', async () => { test('list.splice removes elements and mutates array', async () => {
await expect(`arr = [1 2 3 4 5]; list.splice arr 1 2; arr`).toEvaluateTo([1, 4, 5], globals) await expect(`arr = [1 2 3 4 5]; list.splice arr 1 2; arr`).toEvaluateTo([1, 4, 5])
}) })
test('list.splice returns removed elements', async () => { test('list.splice returns removed elements', async () => {
await expect(`list.splice [1 2 3 4 5] 1 2`).toEvaluateTo([2, 3], globals) await expect(`list.splice [1 2 3 4 5] 1 2`).toEvaluateTo([2, 3])
}) })
test('list.splice from start', async () => { test('list.splice from start', async () => {
await expect(`list.splice [1 2 3 4 5] 0 2`).toEvaluateTo([1, 2], globals) await expect(`list.splice [1 2 3 4 5] 0 2`).toEvaluateTo([1, 2])
}) })
test('list.splice to end', async () => { test('list.splice to end', async () => {
await expect(`arr = [1 2 3 4 5]; list.splice arr 3 2; arr`).toEvaluateTo([1, 2, 3], globals) await expect(`arr = [1 2 3 4 5]; list.splice arr 3 2; arr`).toEvaluateTo([1, 2, 3])
}) })
test('list.sort with no callback sorts ascending', async () => { test('list.sort with no callback sorts ascending', async () => {
await expect(`list.sort [3 1 4 1 5] null`).toEvaluateTo([1, 1, 3, 4, 5], globals) await expect(`list.sort [3 1 4 1 5] null`).toEvaluateTo([1, 1, 3, 4, 5])
}) })
test('list.sort with callback sorts using comparator', async () => { test('list.sort with callback sorts using comparator', async () => {
@ -349,7 +349,7 @@ describe('collections', () => {
b - a b - a
end end
list.sort [3 1 4 1 5] desc list.sort [3 1 4 1 5] desc
`).toEvaluateTo([5, 4, 3, 1, 1], globals) `).toEvaluateTo([5, 4, 3, 1, 1])
}) })
test('list.sort with callback for strings by length', async () => { test('list.sort with callback for strings by length', async () => {
@ -358,52 +358,52 @@ describe('collections', () => {
(length a) - (length b) (length a) - (length b)
end end
list.sort ['cat' 'a' 'dog' 'elephant'] by-length list.sort ['cat' 'a' 'dog' 'elephant'] by-length
`).toEvaluateTo(['a', 'cat', 'dog', 'elephant'], globals) `).toEvaluateTo(['a', 'cat', 'dog', 'elephant'])
}) })
test('list.any? checks if any element matches', async () => { test('list.any? checks if any element matches', async () => {
await expect(` await expect(`
gt-three = do x: x > 3 end gt-three = do x: x > 3 end
list.any? [1 2 4 5] gt-three list.any? [1 2 4 5] gt-three
`).toEvaluateTo(true, globals) `).toEvaluateTo(true)
await expect(` await expect(`
gt-ten = do x: x > 10 end gt-ten = do x: x > 10 end
list.any? [1 2 3] gt-ten list.any? [1 2 3] gt-ten
`).toEvaluateTo(false, globals) `).toEvaluateTo(false)
}) })
test('list.all? checks if all elements match', async () => { test('list.all? checks if all elements match', async () => {
await expect(` await expect(`
positive = do x: x > 0 end positive = do x: x > 0 end
list.all? [1 2 3] positive list.all? [1 2 3] positive
`).toEvaluateTo(true, globals) `).toEvaluateTo(true)
await expect(` await expect(`
positive = do x: x > 0 end positive = do x: x > 0 end
list.all? [1 -2 3] positive list.all? [1 -2 3] positive
`).toEvaluateTo(false, globals) `).toEvaluateTo(false)
}) })
test('list.sum adds all numbers', async () => { test('list.sum adds all numbers', async () => {
await expect(`list.sum [1 2 3 4]`).toEvaluateTo(10, globals) await expect(`list.sum [1 2 3 4]`).toEvaluateTo(10)
await expect(`list.sum []`).toEvaluateTo(0, globals) await expect(`list.sum []`).toEvaluateTo(0)
}) })
test('list.count counts matching elements', async () => { test('list.count counts matching elements', async () => {
await expect(` await expect(`
gt-two = do x: x > 2 end gt-two = do x: x > 2 end
list.count [1 2 3 4 5] gt-two list.count [1 2 3 4 5] gt-two
`).toEvaluateTo(3, globals) `).toEvaluateTo(3)
}) })
test('list.partition splits array by predicate', async () => { test('list.partition splits array by predicate', async () => {
await expect(` await expect(`
gt-two = do x: x > 2 end gt-two = do x: x > 2 end
list.partition [1 2 3 4 5] gt-two list.partition [1 2 3 4 5] gt-two
`).toEvaluateTo([[3, 4, 5], [1, 2]], globals) `).toEvaluateTo([[3, 4, 5], [1, 2]])
}) })
test('list.compact removes null values', async () => { test('list.compact removes null values', async () => {
await expect(`list.compact [1 null 2 null 3]`).toEvaluateTo([1, 2, 3], globals) await expect(`list.compact [1 null 2 null 3]`).toEvaluateTo([1, 2, 3])
}) })
test('list.group-by groups by key function', async () => { test('list.group-by groups by key function', async () => {
@ -416,7 +416,7 @@ describe('collections', () => {
end end
end end
list.group-by ['a' 1 'b' 2] get-type list.group-by ['a' 1 'b' 2] get-type
`).toEvaluateTo({ str: ['a', 'b'], num: [1, 2] }, globals) `).toEvaluateTo({ str: ['a', 'b'], num: [1, 2] })
}) })
}) })
@ -425,14 +425,14 @@ describe('enumerables', () => {
await expect(` await expect(`
double = do x: x * 2 end double = do x: x * 2 end
list.map [1 2 3] double list.map [1 2 3] double
`).toEvaluateTo([2, 4, 6], globals) `).toEvaluateTo([2, 4, 6])
}) })
test('map handles empty array', async () => { test('map handles empty array', async () => {
await expect(` await expect(`
double = do x: x * 2 end double = do x: x * 2 end
list.map [] double list.map [] double
`).toEvaluateTo([], globals) `).toEvaluateTo([])
}) })
test('each iterates over array', async () => { test('each iterates over array', async () => {
@ -441,165 +441,146 @@ describe('enumerables', () => {
await expect(` await expect(`
double = do x: x * 2 end double = do x: x * 2 end
each [1 2 3] double each [1 2 3] double
`).toEvaluateTo([1, 2, 3], globals) `).toEvaluateTo([1, 2, 3])
}) })
test('each handles empty array', async () => { test('each handles empty array', async () => {
await expect(` await expect(`
fn = do x: x end fn = do x: x end
each [] fn each [] fn
`).toEvaluateTo([], globals) `).toEvaluateTo([])
}) })
}) })
describe('dict operations', () => { describe('dict operations', () => {
test('dict.keys returns all keys', async () => { test('dict.keys returns all keys', async () => {
const result = await (async () => { await expect(`dict.keys [a=1 b=2 c=3] | list.sort`).toEvaluateTo(['a', 'b', 'c'].sort())
const { Compiler } = await import('#compiler/compiler')
const { run, fromValue } = await import('reefvm')
const { setGlobals } = await import('#parser/tokenizer')
setGlobals(Object.keys(globals))
const c = new Compiler('dict.keys [a=1 b=2 c=3]')
const r = await run(c.bytecode, globals)
return fromValue(r)
})()
// Check that all expected keys are present (order may vary)
expect(result.sort()).toEqual(['a', 'b', 'c'])
}) })
test('dict.values returns all values', async () => { test('dict.values returns all values', async () => {
const result = await (async () => { await expect('dict.values [a=1 b=2] | list.sort').toEvaluateTo([1, 2].sort())
const { Compiler } = await import('#compiler/compiler')
const { run, fromValue } = await import('reefvm')
const { setGlobals } = await import('#parser/tokenizer')
setGlobals(Object.keys(globals))
const c = new Compiler('dict.values [a=1 b=2]')
const r = await run(c.bytecode, globals)
return fromValue(r)
})()
// Check that all expected values are present (order may vary)
expect(result.sort()).toEqual([1, 2])
}) })
test('dict.has? checks for key', async () => { test('dict.has? checks for key', async () => {
await expect(`dict.has? [a=1 b=2] 'a'`).toEvaluateTo(true, globals) await expect(`dict.has? [a=1 b=2] 'a'`).toEvaluateTo(true)
await expect(`dict.has? [a=1 b=2] 'c'`).toEvaluateTo(false, globals) await expect(`dict.has? [a=1 b=2] 'c'`).toEvaluateTo(false)
}) })
test('dict.get retrieves value with default', async () => { test('dict.get retrieves value with default', async () => {
await expect(`dict.get [a=1] 'a' 0`).toEvaluateTo(1, globals) await expect(`dict.get [a=1] 'a' 0`).toEvaluateTo(1)
await expect(`dict.get [a=1] 'b' 99`).toEvaluateTo(99, globals) await expect(`dict.get [a=1] 'b' 99`).toEvaluateTo(99)
await expect(`dict.get [a=1] 'b'`).toEvaluateTo(null)
}) })
test('dict.set sets value', async () => { test('dict.set sets value', async () => {
await expect(`map = [a=1]; dict.set map 'b' 99; map.b`).toEvaluateTo(99, globals) await expect(`map = [a=1]; dict.set map 'b' 99; map.b`).toEvaluateTo(99)
await expect(`map = [a=1]; dict.set map 'a' 100; map.a`).toEvaluateTo(100, globals) await expect(`map = [a=1]; dict.set map 'a' 100; map.a`).toEvaluateTo(100)
}) })
test('dict.empty? checks if dict is empty', async () => { test('dict.empty? checks if dict is empty', async () => {
await expect(`dict.empty? [=]`).toEvaluateTo(true, globals) await expect(`dict.empty? [=]`).toEvaluateTo(true)
await expect(`dict.empty? [a=1]`).toEvaluateTo(false, globals) await expect(`dict.empty? [a=1]`).toEvaluateTo(false)
}) })
test('dict.merge combines dicts', async () => { test('dict.merge combines dicts', async () => {
await expect(`dict.merge [a=1] [b=2]`).toEvaluateTo({ a: 1, b: 2 }, globals) await expect(`dict.merge [a=1] [b=2]`).toEvaluateTo({ a: 1, b: 2 })
}) })
test('dict.map transforms values', async () => { test('dict.map transforms values', async () => {
await expect(` await expect(`
double = do v k: v * 2 end double = do v k: v * 2 end
dict.map [a=1 b=2] double dict.map [a=1 b=2] double
`).toEvaluateTo({ a: 2, b: 4 }, globals) `).toEvaluateTo({ a: 2, b: 4 })
}) })
test('dict.filter keeps matching entries', async () => { test('dict.filter keeps matching entries', async () => {
await expect(` await expect(`
gt-one = do v k: v > 1 end gt-one = do v k: v > 1 end
dict.filter [a=1 b=2 c=3] gt-one dict.filter [a=1 b=2 c=3] gt-one
`).toEvaluateTo({ b: 2, c: 3 }, globals) `).toEvaluateTo({ b: 2, c: 3 })
}) })
test('dict.from-entries creates dict from array', async () => { test('dict.from-entries creates dict from array', async () => {
await expect(`dict.from-entries [['a' 1] ['b' 2]]`).toEvaluateTo({ a: 1, b: 2 }, globals) await expect(`dict.from-entries [['a' 1] ['b' 2]]`).toEvaluateTo({ a: 1, b: 2 })
}) })
}) })
describe('math operations', () => { describe('math operations', () => {
test('math.abs returns absolute value', async () => { test('math.abs returns absolute value', async () => {
await expect(`math.abs -5`).toEvaluateTo(5, globals) await expect(`math.abs -5`).toEvaluateTo(5)
await expect(`math.abs 5`).toEvaluateTo(5, globals) await expect(`math.abs 5`).toEvaluateTo(5)
}) })
test('math.floor rounds down', async () => { test('math.floor rounds down', async () => {
await expect(`math.floor 3.7`).toEvaluateTo(3, globals) await expect(`math.floor 3.7`).toEvaluateTo(3)
}) })
test('math.ceil rounds up', async () => { test('math.ceil rounds up', async () => {
await expect(`math.ceil 3.2`).toEvaluateTo(4, globals) await expect(`math.ceil 3.2`).toEvaluateTo(4)
}) })
test('math.round rounds to nearest', async () => { test('math.round rounds to nearest', async () => {
await expect(`math.round 3.4`).toEvaluateTo(3, globals) await expect(`math.round 3.4`).toEvaluateTo(3)
await expect(`math.round 3.6`).toEvaluateTo(4, globals) await expect(`math.round 3.6`).toEvaluateTo(4)
}) })
test('math.min returns minimum', async () => { test('math.min returns minimum', async () => {
await expect(`math.min 5 2 8 1`).toEvaluateTo(1, globals) await expect(`math.min 5 2 8 1`).toEvaluateTo(1)
}) })
test('math.max returns maximum', async () => { test('math.max returns maximum', async () => {
await expect(`math.max 5 2 8 1`).toEvaluateTo(8, globals) await expect(`math.max 5 2 8 1`).toEvaluateTo(8)
}) })
test('math.pow computes power', async () => { test('math.pow computes power', async () => {
await expect(`math.pow 2 3`).toEvaluateTo(8, globals) await expect(`math.pow 2 3`).toEvaluateTo(8)
}) })
test('math.sqrt computes square root', async () => { test('math.sqrt computes square root', async () => {
await expect(`math.sqrt 16`).toEvaluateTo(4, globals) await expect(`math.sqrt 16`).toEvaluateTo(4)
}) })
test('math.even? checks if even', async () => { test('math.even? checks if even', async () => {
await expect(`math.even? 4`).toEvaluateTo(true, globals) await expect(`math.even? 4`).toEvaluateTo(true)
await expect(`math.even? 5`).toEvaluateTo(false, globals) await expect(`math.even? 5`).toEvaluateTo(false)
}) })
test('math.odd? checks if odd', async () => { test('math.odd? checks if odd', async () => {
await expect(`math.odd? 5`).toEvaluateTo(true, globals) await expect(`math.odd? 5`).toEvaluateTo(true)
await expect(`math.odd? 4`).toEvaluateTo(false, globals) await expect(`math.odd? 4`).toEvaluateTo(false)
}) })
test('math.positive? checks if positive', async () => { test('math.positive? checks if positive', async () => {
await expect(`math.positive? 5`).toEvaluateTo(true, globals) await expect(`math.positive? 5`).toEvaluateTo(true)
await expect(`math.positive? -5`).toEvaluateTo(false, globals) await expect(`math.positive? -5`).toEvaluateTo(false)
await expect(`math.positive? 0`).toEvaluateTo(false, globals) await expect(`math.positive? 0`).toEvaluateTo(false)
}) })
test('math.negative? checks if negative', async () => { test('math.negative? checks if negative', async () => {
await expect(`math.negative? -5`).toEvaluateTo(true, globals) await expect(`math.negative? -5`).toEvaluateTo(true)
await expect(`math.negative? 5`).toEvaluateTo(false, globals) await expect(`math.negative? 5`).toEvaluateTo(false)
}) })
test('math.zero? checks if zero', async () => { test('math.zero? checks if zero', async () => {
await expect(`math.zero? 0`).toEvaluateTo(true, globals) await expect(`math.zero? 0`).toEvaluateTo(true)
await expect(`math.zero? 5`).toEvaluateTo(false, globals) await expect(`math.zero? 5`).toEvaluateTo(false)
}) })
test('math.clamp restricts value to range', async () => { test('math.clamp restricts value to range', async () => {
await expect(`math.clamp 5 0 10`).toEvaluateTo(5, globals) await expect(`math.clamp 5 0 10`).toEvaluateTo(5)
await expect(`math.clamp -5 0 10`).toEvaluateTo(0, globals) await expect(`math.clamp -5 0 10`).toEvaluateTo(0)
await expect(`math.clamp 15 0 10`).toEvaluateTo(10, globals) await expect(`math.clamp 15 0 10`).toEvaluateTo(10)
}) })
test('math.sign returns sign of number', async () => { test('math.sign returns sign of number', async () => {
await expect(`math.sign 5`).toEvaluateTo(1, globals) await expect(`math.sign 5`).toEvaluateTo(1)
await expect(`math.sign -5`).toEvaluateTo(-1, globals) await expect(`math.sign -5`).toEvaluateTo(-1)
await expect(`math.sign 0`).toEvaluateTo(0, globals) await expect(`math.sign 0`).toEvaluateTo(0)
}) })
test('math.trunc truncates decimal', async () => { test('math.trunc truncates decimal', async () => {
await expect(`math.trunc 3.7`).toEvaluateTo(3, globals) await expect(`math.trunc 3.7`).toEvaluateTo(3)
await expect(`math.trunc -3.7`).toEvaluateTo(-3, globals) await expect(`math.trunc -3.7`).toEvaluateTo(-3)
}) })
}) })