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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
|
from collections import OrderedDict
from hypothesis import given
from hypothesis.strategies import floats, integers
import numpy
import pytest
import snuggs
@pytest.fixture
def ones():
return numpy.ones((2, 2))
@pytest.fixture
def truetrue():
return numpy.array([True, True])
@pytest.fixture
def truefalse():
return numpy.array([True, False])
@given(integers())
def test_integer_operand(num):
assert list(snuggs.operand.parseString(repr(num))) == [num]
@given(floats(allow_infinity=False, allow_nan=False))
def test_real_operand(num):
assert list(snuggs.operand.parseString(repr(num))) == [num]
def test_int_expr():
assert snuggs.eval('(+ 1 2)') == 3
def test_int_mult_expr():
assert snuggs.eval('(+ 1 2 3)') == 6
def test_real_expr():
assert round(snuggs.eval('(* 0.1 0.2)'), 3) == 0.02
def test_int_real_expr():
assert snuggs.eval('(+ 2 1.1)') == 3.1
def test_real_int_expr():
assert snuggs.eval('(+ 1.1 2)') == 3.1
def test_arr_var(ones):
r = snuggs.eval('(+ foo 0)', foo=ones)
assert list(r.flatten()) == [1, 1, 1, 1]
def test_arr_lookup(ones):
kwargs = OrderedDict((('foo', ones),
('bar', 2.0 * ones),
('a', 3.0 * ones)))
r = snuggs.eval('(read 1)', kwargs)
assert list(r.flatten()) == [1, 1, 1, 1]
def test_arr_var_long(ones):
r = snuggs.eval('(+ FOO_BAR_42 0)', FOO_BAR_42=ones)
assert list(r.flatten()) == [1, 1, 1, 1]
@pytest.mark.xfail(reason="Keyword argument order can't be relied on")
def test_arr_lookup_kwarg_order(ones):
kwargs = OrderedDict((('foo', ones),
('bar', 2.0 * ones),
('a', 3.0 * ones)))
r = snuggs.eval('(read 1)', **kwargs)
assert list(r.flatten()) == [1, 1, 1, 1]
def test_arr_lookup_2(ones):
r = snuggs.eval('(read 1 1)', foo=ones)
assert list(r.flatten()) == [1, 1]
def test_arr_take(ones):
r = snuggs.eval('(take foo 1)', foo=ones)
assert list(r.flatten()) == [1, 1]
r = snuggs.eval('(take foo 2)', foo=ones)
assert list(r.flatten()) == [1, 1]
def test_int_arr_expr(ones):
result = snuggs.eval('(+ foo 1)', foo=ones)
assert list(result.flatten()) == [2, 2, 2, 2]
def test_int_arr_expr_by_name(ones):
result = snuggs.eval('(+ (read 1) 1.5)', foo=ones)
assert list(result.flatten()) == [2.5, 2.5, 2.5, 2.5]
def test_int_arr_read(ones):
result = snuggs.eval('(+ (read 1 1) 1.5)', foo=ones)
assert list(result.flatten()) == [2.5, 2.5]
def test_list(ones):
result = snuggs.eval(
'(asarray (take foo 1) (take foo 1) (take bar 1) (take bar 1))',
foo=ones, bar=ones)
assert list(result.flatten()) == [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
def test_eq(ones):
ones[0][0] = 2
result = snuggs.eval('(== foo 1)', foo=ones)
assert list(result.flatten()) == [False, True, True, True]
def test_or(truetrue, truefalse):
result = snuggs.eval(
'(| foo bar)', foo=truetrue, bar=truefalse)
assert list(result.flatten()) == [True, True]
def test_and(truetrue, truefalse):
result = snuggs.eval(
'(& foo bar)', foo=truetrue, bar=truefalse)
assert list(result.flatten()) == [True, False]
def test_ones_like(truefalse):
result = snuggs.eval("(ones_like foo 'uint8')", foo=truefalse)
assert list(result.flatten()) == [1.0, 1.0]
def test_full_like(truefalse):
result = snuggs.eval("(full_like foo 3.14 'float64')", foo=truefalse)
assert list(result.flatten()) == [3.14, 3.14]
result = snuggs.eval('(full_like foo 3.14 "float64")', foo=truefalse)
assert list(result.flatten()) == [3.14, 3.14]
def test_ufunc(truetrue, truefalse):
result = snuggs.eval(
'(where (& foo bar) 1 0)', foo=truetrue, bar=truefalse)
assert list(result.flatten()) == [1.0, 0.0]
def test_partial():
result = snuggs.eval('((partial * 2) 2)')
assert result == 4
def test_map_func():
result = snuggs.eval('(map sqrt (asarray 1 4 9))')
assert list(result) == [1, 2, 3]
def test_map_partial():
result = snuggs.eval('(map (partial * 2) (asarray 1 2 3))')
assert list(result) == [2, 4, 6]
def test_map_asarray():
result = snuggs.eval('(asarray (map (partial * 2) (asarray 1 2 3)))')
assert list(result) == [2, 4, 6]
def test_multi_operator_array(ones):
result = snuggs.eval(
'(+ ones (/ ones 1 0.5) (* ones 1 3))', ones=ones)
assert list(result.flatten()) == [6.0] * 4
def test_nil():
assert snuggs.eval('(== nil nil)')
assert not snuggs.eval('(== 1 nil)')
assert not snuggs.eval('(== nil 1)')
assert snuggs.eval('(!= 1 nil)')
assert snuggs.eval('(!= nil 1)')
def test_masked_arr():
foo = numpy.ma.masked_equal(numpy.array([0, 0, 0, 1], dtype='uint8'), 0)
r = snuggs.eval('(+ foo 1)', foo=foo)
assert list(r.data.flatten()) == [0, 0, 0, 2]
assert list(r.flatten()) == [numpy.ma.masked, numpy.ma.masked, numpy.ma.masked, 2]
# Parse and syntax error testing.
def test_missing_closing_paren():
with pytest.raises(SyntaxError) as excinfo:
snuggs.eval("(+ 1 2")
assert excinfo.value.lineno == 1
assert excinfo.value.offset == 7
def test_missing_func():
with pytest.raises(SyntaxError) as excinfo:
snuggs.eval("(0 1 2)")
assert excinfo.value.lineno == 1
assert excinfo.value.offset == 2
assert str(excinfo.value) == "'0' is not a function or operator"
def test_missing_func2():
with pytest.raises(SyntaxError) as excinfo:
snuggs.eval("(# 1 2)")
assert excinfo.value.lineno == 1
assert excinfo.value.offset == 2
def test_undefined_var():
with pytest.raises(SyntaxError) as excinfo:
snuggs.eval("(+ 1 bogus)")
assert excinfo.value.lineno == 1
assert excinfo.value.offset == 6
assert str(excinfo.value) == "name 'bogus' is not defined"
def test_bogus_higher_order_func():
with pytest.raises(SyntaxError) as excinfo:
snuggs.eval("((bogus * 2) 2)")
assert excinfo.value.lineno == 1
assert excinfo.value.offset == 3
def test_type_error():
with pytest.raises(TypeError):
snuggs.eval("(+ 1 'bogus')")
def test_negative_decimal():
"""Negative decimals parse correctly"""
assert snuggs.eval("(< -0.9 0)")
|
