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
|
from py.test import raises
from pypy.module.micronumpy import support
from pypy.module.micronumpy.ufuncs import W_UfuncGeneric
from pypy.module.micronumpy.test.test_base import BaseNumpyAppTest
from pypy.interpreter.error import OperationError
class TestParseSignatureDirect(BaseNumpyAppTest):
def test_signature_basic(self):
space = self.space
funcs = [None]
name = 'dummy ufunc'
identity = None
dtypes = [int, int, int]
nin = 2
nout = 1
signature = '(), () -> ( ) '
ufunc = W_UfuncGeneric(space, funcs, name, identity, nin, nout, dtypes, signature)
# make sure no attributes are added
attribs = set(ufunc.__dict__.keys())
support._parse_signature(space, ufunc, ufunc.signature)
new_attribs = set(ufunc.__dict__.keys())
assert attribs == new_attribs
assert sum(ufunc.core_num_dims) == 0
assert ufunc.core_enabled == 0
nin = 2
nout = 1
signature = '(i),(i)->()'
ufunc = W_UfuncGeneric(space, funcs, name, identity, nin, nout, dtypes, signature)
support._parse_signature(space, ufunc, ufunc.signature)
assert ufunc.core_enabled == 1
nin = 2
nout = 1
signature = '(i1, i2),(J_1)->(_kAB)'
ufunc = W_UfuncGeneric(space, funcs, name, identity, nin, nout, dtypes, signature)
support._parse_signature(space, ufunc, ufunc.signature)
assert ufunc.core_enabled == 1
nin = 2
nout = 1
signature = '(i1 i2),(J_1)->(_kAB)'
ufunc = W_UfuncGeneric(space, funcs, name, identity, nin, nout, dtypes, signature)
exc = raises(OperationError, support._parse_signature, space, ufunc, ufunc.signature)
assert "expect dimension name" in exc.value.errorstr(space)
nin = 2
nout = 1
signature = '(i),i(->()'
ufunc = W_UfuncGeneric(space, funcs, name, identity, nin, nout, dtypes, signature)
exc = raises(OperationError, support._parse_signature, space, ufunc, ufunc.signature)
assert "expect '(' at 4" in exc.value.errorstr(space)
|