196 lines
3.7 KiB
Python
196 lines
3.7 KiB
Python
"""
|
|
Doctests for `parse`
|
|
--------------------
|
|
|
|
# tag::PARSE[]
|
|
>>> from lis import parse
|
|
>>> parse('1.5')
|
|
1.5
|
|
>>> parse('ni!')
|
|
'ni!'
|
|
>>> parse('(gcd 18 45)')
|
|
['gcd', 18, 45]
|
|
>>> parse('''
|
|
... (define double
|
|
... (lambda (n)
|
|
... (* n 2)))
|
|
... ''')
|
|
['define', 'double', ['lambda', ['n'], ['*', 'n', 2]]]
|
|
|
|
# end::PARSE[]
|
|
|
|
Doctest for `Environment`
|
|
-------------------------
|
|
|
|
# tag::ENVIRONMENT[]
|
|
>>> from lis import Environment
|
|
>>> inner_env = {'a': 2}
|
|
>>> outer_env = {'a': 0, 'b': 1}
|
|
>>> env = Environment(inner_env, outer_env)
|
|
>>> env['a'] = 111 # <1>
|
|
>>> env['c'] = 222
|
|
>>> env
|
|
Environment({'a': 111, 'c': 222}, {'a': 0, 'b': 1})
|
|
>>> env.change('b', 333) # <2>
|
|
>>> env
|
|
Environment({'a': 111, 'c': 222}, {'a': 0, 'b': 333})
|
|
|
|
# end::ENVIRONMENT[]
|
|
|
|
Doctests for `evaluate`
|
|
-----------------------
|
|
|
|
# tag::EVAL_NUMBER[]
|
|
>>> from lis import parse, evaluate, standard_env
|
|
>>> evaluate(parse('1.5'), {})
|
|
1.5
|
|
|
|
# end::EVAL_NUMBER[]
|
|
|
|
# tag::EVAL_SYMBOL[]
|
|
>>> from lis import standard_env
|
|
>>> evaluate(parse('+'), standard_env())
|
|
<built-in function add>
|
|
>>> evaluate(parse('ni!'), standard_env())
|
|
Traceback (most recent call last):
|
|
...
|
|
KeyError: 'ni!'
|
|
|
|
# end::EVAL_SYMBOL[]
|
|
|
|
|
|
# tag::EVAL_QUOTE[]
|
|
>>> evaluate(parse('(quote no-such-name)'), standard_env())
|
|
'no-such-name'
|
|
>>> evaluate(parse('(quote (99 bottles of beer))'), standard_env())
|
|
[99, 'bottles', 'of', 'beer']
|
|
>>> evaluate(parse('(quote (/ 10 0))'), standard_env())
|
|
['/', 10, 0]
|
|
|
|
# end::EVAL_QUOTE[]
|
|
|
|
# tag::EVAL_IF[]
|
|
>>> evaluate(parse('(if (= 3 3) 1 0))'), standard_env())
|
|
1
|
|
>>> evaluate(parse('(if (= 3 4) 1 0))'), standard_env())
|
|
0
|
|
|
|
# end::EVAL_IF[]
|
|
|
|
|
|
# tag::EVAL_LAMBDA[]
|
|
>>> expr = '(lambda (a b) (* (/ a b) 100))'
|
|
>>> f = evaluate(parse(expr), standard_env())
|
|
>>> f # doctest: +ELLIPSIS
|
|
<lis.Procedure object at 0x...>
|
|
>>> f(15, 20)
|
|
75.0
|
|
|
|
# end::EVAL_LAMBDA[]
|
|
|
|
# tag::EVAL_DEFINE[]
|
|
>>> global_env = standard_env()
|
|
>>> evaluate(parse('(define answer (* 7 6))'), global_env)
|
|
>>> global_env['answer']
|
|
42
|
|
|
|
# end::EVAL_DEFINE[]
|
|
|
|
# tag::EVAL_DEFUN[]
|
|
>>> global_env = standard_env()
|
|
>>> percent = '(define % (lambda (a b) (* (/ a b) 100)))'
|
|
>>> evaluate(parse(percent), global_env)
|
|
>>> global_env['%'] # doctest: +ELLIPSIS
|
|
<lis.Procedure object at 0x...>
|
|
>>> global_env['%'](170, 200)
|
|
85.0
|
|
|
|
# end::EVAL_DEFUN[]
|
|
|
|
function call:
|
|
|
|
# tag::EVAL_CALL[]
|
|
>>> evaluate(parse('(% (* 12 14) (- 500 100))'), global_env)
|
|
42.0
|
|
|
|
# end::EVAL_CALL[]
|
|
|
|
"""
|
|
|
|
import math
|
|
|
|
from lis import run
|
|
|
|
|
|
fact_src = """
|
|
(define !
|
|
(lambda (n)
|
|
(if (< n 2)
|
|
1
|
|
(* n (! (- n 1)))
|
|
)
|
|
)
|
|
)
|
|
(! 42)
|
|
"""
|
|
def test_factorial():
|
|
got = run(fact_src)
|
|
assert got == 1405006117752879898543142606244511569936384000000000
|
|
assert got == math.factorial(42)
|
|
|
|
closure_src = """
|
|
(define make-adder
|
|
(lambda (increment)
|
|
(lambda (x) (+ increment x))
|
|
)
|
|
)
|
|
(define inc (make-adder 1))
|
|
(inc 99)
|
|
"""
|
|
def test_closure():
|
|
got = run(closure_src)
|
|
assert got == 100
|
|
|
|
closure_with_change_src = """
|
|
(define make-counter
|
|
(lambda ()
|
|
(define n 0)
|
|
(lambda ()
|
|
(set! n (+ n 1))
|
|
n)
|
|
)
|
|
)
|
|
(define counter (make-counter))
|
|
(display (counter))
|
|
(display (counter))
|
|
(display (counter))
|
|
"""
|
|
def test_closure_with_change(capsys):
|
|
run(closure_with_change_src)
|
|
captured = capsys.readouterr()
|
|
assert captured.out == '1\n2\n3\n'
|
|
|
|
|
|
|
|
# tag::RUN_AVERAGER[]
|
|
closure_averager_src = """
|
|
(define make-averager
|
|
(lambda ()
|
|
(define count 0)
|
|
(define total 0)
|
|
(lambda (new-value)
|
|
(set! count (+ count 1))
|
|
(set! total (+ total new-value))
|
|
(/ total count)
|
|
)
|
|
)
|
|
)
|
|
(define avg (make-averager))
|
|
(avg 10)
|
|
(avg 11)
|
|
(avg 15)
|
|
"""
|
|
def test_closure_averager():
|
|
got = run(closure_averager_src)
|
|
assert got == 12.0
|
|
# end::RUN_AVERAGER[] |