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
|
import os
import math
from numpy.testing import *
from numpy import array
import util
def _path(*a):
return os.path.join(*((os.path.dirname(__file__),) + a))
class TestSizeSumExample(util.F2PyTest):
sources = [_path('src', 'size', 'foo.f90'),
]
@dec.slow
def test_all(self):
r = self.module.foo([[1,2]])
assert_equal(r, [3],`r`)
r = self.module.foo([[1,2],[3,4]])
assert_equal(r, [3,7],`r`)
r = self.module.foo([[1,2],[3,4],[5,6]])
assert_equal(r, [3,7,11],`r`)
@dec.slow
def test_transpose(self):
r = self.module.trans([[1,2]])
assert_equal(r, [[1],[2]],`r`)
r = self.module.trans([[1,2,3],[4,5,6]])
assert_equal(r, [[1,4],[2,5],[3,6]],`r`)
@dec.slow
def test_flatten(self):
r = self.module.flatten([[1,2]])
assert_equal(r, [1,2],`r`)
r = self.module.flatten([[1,2,3],[4,5,6]])
assert_equal(r, [1,2,3,4,5,6],`r`)
if __name__ == "__main__":
import nose
nose.runmodule()
|
