|
1 | 1 | from typing import Optional |
2 | 2 |
|
3 | | -from pytest import mark |
| 3 | +from pytest import mark, fixture |
4 | 4 |
|
5 | 5 | from lis import parse, evaluate, Expression, Environment, standard_env |
6 | 6 |
|
|
39 | 39 | ("((repeat riff-shuffle) (list 1 2 3 4 5 6 7 8))", [1, 3, 5, 7, 2, 4, 6, 8]), |
40 | 40 | ("(riff-shuffle (riff-shuffle (riff-shuffle (list 1 2 3 4 5 6 7 8))))", [1,2,3,4,5,6,7,8]), |
41 | 41 | ]) |
42 | | -@mark.skip |
43 | 42 | def test_evaluate(source: str, expected: Optional[Expression]) -> None: |
44 | 43 | got = evaluate(parse(source)) |
45 | 44 | assert got == expected |
46 | 45 |
|
47 | 46 |
|
| 47 | +@fixture |
| 48 | +def std_env() -> Environment: |
| 49 | + return standard_env() |
| 50 | + |
48 | 51 | # tests for each of the cases in evaluate |
49 | 52 |
|
50 | 53 | def test_evaluate_variable() -> None: |
@@ -83,65 +86,58 @@ def test_evaluate_if_false() -> None: |
83 | 86 | assert got == expected |
84 | 87 |
|
85 | 88 |
|
86 | | -def test_define() -> None: |
87 | | - env: Environment = standard_env() |
| 89 | +def test_define(std_env: Environment) -> None: |
88 | 90 | source = '(define answer (* 6 7))' |
89 | | - got = evaluate(parse(source), env) |
| 91 | + got = evaluate(parse(source), std_env) |
90 | 92 | assert got is None |
91 | | - assert env['answer'] == 42 |
| 93 | + assert std_env['answer'] == 42 |
92 | 94 |
|
93 | 95 |
|
94 | | -def test_lambda() -> None: |
95 | | - env: Environment = standard_env() |
| 96 | +def test_lambda(std_env: Environment) -> None: |
96 | 97 | source = '(lambda (a b) (if (>= a b) a b))' |
97 | | - func = evaluate(parse(source), env) |
| 98 | + func = evaluate(parse(source), std_env) |
98 | 99 | assert func.parms == ['a', 'b'] |
99 | 100 | assert func.body == ['if', ['>=', 'a', 'b'], 'a', 'b'] |
100 | | - assert func.env is env |
| 101 | + assert func.env is std_env |
101 | 102 | assert func(1, 2) == 2 |
102 | 103 | assert func(3, 2) == 3 |
103 | 104 |
|
104 | 105 |
|
105 | | -def test_begin() -> None: |
106 | | - env: Environment = standard_env() |
| 106 | +def test_begin(std_env: Environment) -> None: |
107 | 107 | source = """ |
108 | 108 | (begin |
109 | 109 | (define x (* 2 3)) |
110 | 110 | (* x 7) |
111 | 111 | ) |
112 | 112 | """ |
113 | | - got = evaluate(parse(source), env) |
| 113 | + got = evaluate(parse(source), std_env) |
114 | 114 | assert got == 42 |
115 | 115 |
|
116 | 116 |
|
117 | | -def test_invocation_builtin_car() -> None: |
118 | | - env: Environment = standard_env() |
| 117 | +def test_invocation_builtin_car(std_env: Environment) -> None: |
119 | 118 | source = '(car (quote (11 22 33)))' |
120 | | - got = evaluate(parse(source), env) |
| 119 | + got = evaluate(parse(source), std_env) |
121 | 120 | assert got == 11 |
122 | 121 |
|
123 | 122 |
|
124 | | -def test_invocation_builtin_append() -> None: |
125 | | - env: Environment = standard_env() |
| 123 | +def test_invocation_builtin_append(std_env: Environment) -> None: |
126 | 124 | source = '(append (quote (a b)) (quote (c d)))' |
127 | | - got = evaluate(parse(source), env) |
| 125 | + got = evaluate(parse(source), std_env) |
128 | 126 | assert got == ['a', 'b', 'c', 'd'] |
129 | 127 |
|
130 | 128 |
|
131 | | -def test_invocation_builtin_map() -> None: |
132 | | - env: Environment = standard_env() |
| 129 | +def test_invocation_builtin_map(std_env: Environment) -> None: |
133 | 130 | source = '(map (lambda (x) (* x 2)) (quote (1 2 3))))' |
134 | | - got = evaluate(parse(source), env) |
| 131 | + got = evaluate(parse(source), std_env) |
135 | 132 | assert got == [2, 4, 6] |
136 | 133 |
|
137 | 134 |
|
138 | | -def test_invocation_user_procedure() -> None: |
139 | | - env: Environment = standard_env() |
| 135 | +def test_invocation_user_procedure(std_env: Environment) -> None: |
140 | 136 | source = """ |
141 | 137 | (begin |
142 | 138 | (define max (lambda (a b) (if (>= a b) a b))) |
143 | 139 | (max 22 11) |
144 | 140 | ) |
145 | 141 | """ |
146 | | - got = evaluate(parse(source), env) |
| 142 | + got = evaluate(parse(source), std_env) |
147 | 143 | assert got == 22 |
0 commit comments