1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
(import-macros {: fn*} :macros.fn)
;; requires `eq?' from core.fnl to be available at runtime
(fn* assert-eq
([expr1 expr2]
(assert-eq expr1 expr2 'nil))
([expr1 expr2 msg]
`(let [left# ,expr1
right# ,expr2
view# (require :fennelview)]
(assert (eq? left# right#) (or ,msg (.. "equality assertion failed
Left: " (view# left#) "
Right: " (view# right#) "\n"))))))
(fn* assert-ne
([expr1 expr2]
(assert-ne expr1 expr2 'nil))
([expr1 expr2 msg]
`(let [left# ,expr1
right# ,expr2
view# (require :fennelview)]
(assert (not (eq? left# right#)) (or ,msg (.. "unequality assertion failed
Left: " (view# left#) "
Right: " (view# right#) "\n"))))))
(fn walk-tree [root f custom-iterator]
"Walks a tree (like the AST), invoking f(node, idx, parent) on each node.
When f returns a truthy value, recursively walks the children."
(fn walk [iterfn parent idx node]
(when (f idx node parent)
(each [k v (iterfn node)]
(walk iterfn node k v))))
(walk (or custom-iterator pairs) nil nil root)
root)
(fn* assert*
([expr]
(assert* expr 'nil))
([expr msg]
`(assert ,expr (.. "assertion failed for " (or ,msg ,(tostring expr))))))
(fn* test
;"define test function, print its name and run it."
[name docstring & body]
(let [test-name (sym (.. (tostring name) "-test"))]
`(do (fn ,test-name []
,(or docstring nil)
((or table.unpack unpack) ,body))
(io.stderr:write (.. "running: " ,(tostring test-name) "\n"))
(,test-name))))
{: assert-eq
: assert-ne
: assert*
: test}
|