Files
rose-ash/shared/sx/tests/test_sx_ref.py

416 lines
13 KiB
Python

"""Tests for the transpiled sx_ref.py evaluator.
Runs the same test cases as test_evaluator.py and test_html.py but
against the bootstrap-compiled evaluator to verify correctness.
"""
import pytest
from shared.sx.parser import parse
from shared.sx.types import Symbol, Keyword, NIL, Lambda, Component, Macro, PageDef
from shared.sx.ref import sx_ref
# ---------------------------------------------------------------------------
# Helpers
# ---------------------------------------------------------------------------
def ev(text, env=None):
"""Parse and evaluate a single expression via sx_ref."""
return sx_ref.evaluate(parse(text), env)
def render(text, env=None):
"""Parse and render via sx_ref."""
return sx_ref.render(parse(text), env)
# ---------------------------------------------------------------------------
# Literals and lookups
# ---------------------------------------------------------------------------
class TestLiterals:
def test_int(self):
assert ev("42") == 42
def test_string(self):
assert ev('"hello"') == "hello"
def test_true(self):
assert ev("true") is True
def test_nil(self):
assert ev("nil") is NIL
def test_symbol_lookup(self):
assert ev("x", {"x": 10}) == 10
def test_undefined_symbol(self):
with pytest.raises(sx_ref.EvalError, match="Undefined symbol"):
ev("xyz")
def test_keyword_evaluates_to_name(self):
assert ev(":foo") == "foo"
# ---------------------------------------------------------------------------
# Arithmetic
# ---------------------------------------------------------------------------
class TestArithmetic:
def test_add(self):
assert ev("(+ 1 2 3)") == 6
def test_sub(self):
assert ev("(- 10 3)") == 7
def test_mul(self):
assert ev("(* 2 3 4)") == 24
def test_div(self):
assert ev("(/ 10 4)") == 2.5
def test_mod(self):
assert ev("(mod 7 3)") == 1
# ---------------------------------------------------------------------------
# Special forms
# ---------------------------------------------------------------------------
class TestSpecialForms:
def test_if_true(self):
assert ev("(if true 1 2)") == 1
def test_if_false(self):
assert ev("(if false 1 2)") == 2
def test_if_no_else(self):
assert ev("(if false 1)") is NIL
def test_when_true(self):
assert ev("(when true 42)") == 42
def test_when_false(self):
assert ev("(when false 42)") is NIL
def test_and_short_circuit(self):
assert ev("(and true true 3)") == 3
assert ev("(and true false 3)") is False
def test_or_short_circuit(self):
assert ev("(or false false 3)") == 3
assert ev("(or false 2 3)") == 2
def test_let_scheme_style(self):
assert ev("(let ((x 10) (y 20)) (+ x y))") == 30
def test_let_clojure_style(self):
assert ev("(let (x 10 y 20) (+ x y))") == 30
def test_let_sequential(self):
assert ev("(let ((x 1) (y (+ x 1))) y)") == 2
def test_begin(self):
assert ev("(begin 1 2 3)") == 3
def test_quote(self):
result = ev("(quote (a b c))")
assert result == [Symbol("a"), Symbol("b"), Symbol("c")]
def test_cond_clojure(self):
assert ev("(cond false 1 true 2 :else 3)") == 2
def test_cond_else(self):
assert ev("(cond false 1 false 2 :else 99)") == 99
def test_case(self):
assert ev('(case 2 1 "one" 2 "two" :else "other")') == "two"
def test_thread_first(self):
assert ev("(-> 5 (+ 3) (* 2))") == 16
def test_define(self):
env = {}
ev("(define x 42)", env)
assert env["x"] == 42
# ---------------------------------------------------------------------------
# Lambda
# ---------------------------------------------------------------------------
class TestLambda:
def test_create_and_call(self):
assert ev("((fn (x) (* x x)) 5)") == 25
def test_closure(self):
result = ev("(let ((a 10)) ((fn (x) (+ x a)) 5))")
assert result == 15
def test_higher_order(self):
result = ev("(let ((double (fn (x) (* x 2)))) (double 7))")
assert result == 14
# ---------------------------------------------------------------------------
# Collections
# ---------------------------------------------------------------------------
class TestCollections:
def test_list_constructor(self):
assert ev("(list 1 2 3)") == [1, 2, 3]
def test_dict_constructor(self):
assert ev("(dict :a 1 :b 2)") == {"a": 1, "b": 2}
def test_get_dict(self):
assert ev('(get {:a 1 :b 2} "a")') == 1
def test_get_list(self):
assert ev("(get (list 10 20 30) 1)") == 20
def test_first_last_rest(self):
assert ev("(first (list 1 2 3))") == 1
assert ev("(last (list 1 2 3))") == 3
assert ev("(rest (list 1 2 3))") == [2, 3]
def test_len(self):
assert ev("(len (list 1 2 3))") == 3
def test_concat(self):
assert ev("(concat (list 1 2) (list 3 4))") == [1, 2, 3, 4]
def test_cons(self):
assert ev("(cons 0 (list 1 2))") == [0, 1, 2]
def test_merge(self):
assert ev("(merge {:a 1} {:b 2} {:a 3})") == {"a": 3, "b": 2}
def test_empty(self):
assert ev("(empty? (list))") is True
assert ev("(empty? (list 1))") is False
assert ev("(empty? nil)") is True
# ---------------------------------------------------------------------------
# Higher-order forms
# ---------------------------------------------------------------------------
class TestHigherOrder:
def test_map(self):
assert ev("(map (fn (x) (* x x)) (list 1 2 3 4))") == [1, 4, 9, 16]
def test_filter(self):
assert ev("(filter (fn (x) (> x 2)) (list 1 2 3 4))") == [3, 4]
def test_reduce(self):
assert ev("(reduce (fn (acc x) (+ acc x)) 0 (list 1 2 3))") == 6
def test_some(self):
assert ev("(some (fn (x) (> x 3)) (list 1 2 3 4 5))") is True
def test_for_each(self):
result = ev("(for-each (fn (x) x) (list 1 2 3))")
assert result is NIL
# ---------------------------------------------------------------------------
# String ops
# ---------------------------------------------------------------------------
class TestStrings:
def test_str(self):
assert ev('(str "hello" " " "world")') == "hello world"
def test_upper_lower(self):
assert ev('(upper "hello")') == "HELLO"
assert ev('(lower "HELLO")') == "hello"
def test_split_join(self):
assert ev('(split "a,b,c" ",")') == ["a", "b", "c"]
assert ev('(join "-" (list "a" "b"))') == "a-b"
def test_starts_ends(self):
assert ev('(starts-with? "hello" "hel")') is True
assert ev('(ends-with? "hello" "llo")') is True
# ---------------------------------------------------------------------------
# Components
# ---------------------------------------------------------------------------
class TestComponents:
def test_defcomp_and_render(self):
env = {}
ev("(defcomp ~box (&key title) (div :class \"box\" title))", env)
result = render("(~box :title \"hi\")", env)
assert result == '<div class="box">hi</div>'
def test_defcomp_with_children(self):
env = {}
ev("(defcomp ~wrap (&rest children) (div children))", env)
result = render('(~wrap (span "a") (span "b"))', env)
assert result == '<div><span>a</span><span>b</span></div>'
# ---------------------------------------------------------------------------
# HTML rendering
# ---------------------------------------------------------------------------
class TestHTMLRendering:
def test_basic_element(self):
assert render("(div)") == "<div></div>"
def test_text_content(self):
assert render('(p "hello")') == "<p>hello</p>"
def test_attributes(self):
result = render('(a :href "/about" "link")')
assert result == '<a href="/about">link</a>'
def test_void_element(self):
result = render('(br)')
assert result == "<br />"
def test_nested(self):
result = render('(div (p "a") (p "b"))')
assert result == "<div><p>a</p><p>b</p></div>"
def test_fragment(self):
result = render('(<> (span "a") (span "b"))')
assert result == "<span>a</span><span>b</span>"
def test_conditional_rendering(self):
result = render('(if true (span "yes") (span "no"))')
assert result == "<span>yes</span>"
def test_map_rendering(self):
result = render('(map (fn (x) (li x)) (list "a" "b"))')
assert result == "<li>a</li><li>b</li>"
def test_html_escaping(self):
result = render('(span "<b>bold</b>")')
assert result == "<span>&lt;b&gt;bold&lt;/b&gt;</span>"
# ---------------------------------------------------------------------------
# Aser (SX wire format)
# ---------------------------------------------------------------------------
class TestAser:
def test_render_to_sx_basic(self):
expr = parse("(div :class \"foo\" \"hello\")")
result = sx_ref.render_to_sx(expr, {})
assert result == '(div :class "foo" "hello")'
def test_component_not_expanded(self):
expr = parse('(~card :title "hi")')
result = sx_ref.render_to_sx(expr, {})
assert result == '(~card :title "hi")'
def test_fragment(self):
expr = parse('(<> "a" "b")')
result = sx_ref.render_to_sx(expr, {})
assert result == '(<> "a" "b")'
def test_let_evaluates(self):
expr = parse('(let ((x 5)) x)')
result = sx_ref.render_to_sx(expr, {})
assert result == "5"
def test_if_evaluates(self):
expr = parse('(if true "yes" "no")')
result = sx_ref.render_to_sx(expr, {})
assert result == 'yes' # strings pass through unserialized
# ---------------------------------------------------------------------------
# Macros
# ---------------------------------------------------------------------------
class TestDefcomp:
def test_defcomp_basic(self):
"""defcomp should parse &key params without trying to eval &key as a symbol."""
env = {}
ev('(defcomp ~card (&key title) (div title))', env)
assert isinstance(env.get("~card"), Component)
def test_defcomp_render(self):
env = {}
ev('(defcomp ~card (&key title) (div title))', env)
result = render('(~card :title "hello")', env)
assert result == '<div>hello</div>'
def test_defcomp_with_children(self):
env = {}
ev('(defcomp ~wrap (&key title &rest children) (div title children))', env)
comp = env["~wrap"]
assert comp.has_children is True
assert "title" in comp.params
def test_defcomp_multiple_params(self):
env = {}
ev('(defcomp ~box (&key a b c) (div a b c))', env)
result = render('(~box :a "1" :b "2" :c "3")', env)
assert result == '<div>123</div>'
class TestDefhandler:
def test_defhandler_basic(self):
"""defhandler should parse &key params and create a HandlerDef."""
from shared.sx.types import HandlerDef
env = {}
ev('(defhandler link-card (&key slug keys) (div slug))', env)
hdef = env.get("handler:link-card")
assert isinstance(hdef, HandlerDef)
assert hdef.name == "link-card"
assert hdef.params == ["slug", "keys"]
def test_defquery_basic(self):
from shared.sx.types import QueryDef
env = {}
ev('(defquery get-post (&key slug) "Fetch a post" (list slug))', env)
qdef = env.get("query:get-post")
assert isinstance(qdef, QueryDef)
assert qdef.params == ["slug"]
assert qdef.doc == "Fetch a post"
def test_defaction_basic(self):
from shared.sx.types import ActionDef
env = {}
ev('(defaction save-post (&key title body) (list title body))', env)
adef = env.get("action:save-post")
assert isinstance(adef, ActionDef)
assert adef.params == ["title", "body"]
class TestHOWithNativeCallable:
def test_map_with_native_fn(self):
"""map should work with native callables (primitives), not just Lambda."""
result = ev('(map str (list 1 2 3))')
assert result == ["1", "2", "3"]
def test_filter_with_native_fn(self):
result = ev('(filter number? (list 1 "a" 2 "b" 3))')
assert result == [1, 2, 3]
def test_map_with_env_fn(self):
"""map should work with Python functions registered in env."""
env = {"double": lambda x: x * 2}
result = ev('(map double (list 1 2 3))', env)
assert result == [2, 4, 6]
def test_ho_non_callable_fails_fast(self):
"""Passing a non-callable to map should error clearly."""
import pytest
with pytest.raises(sx_ref.EvalError, match="Not callable"):
ev('(map 42 (list 1 2 3))')
class TestMacros:
def test_defmacro_and_expand(self):
env = {}
ev("(defmacro unless (test body) (list (quote if) (list (quote not) test) body))", env)
result = ev('(unless false "ran")', env)
assert result == "ran"