46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
import os
|
||
|
|
import pytest
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
from . import util
|
||
|
|
|
||
|
|
|
||
|
|
class TestSizeSumExample(util.F2PyTest):
|
||
|
|
sources = [util.getpath("tests", "src", "size", "foo.f90")]
|
||
|
|
|
||
|
|
@pytest.mark.slow
|
||
|
|
def test_all(self):
|
||
|
|
r = self.module.foo([[]])
|
||
|
|
assert r == [0]
|
||
|
|
|
||
|
|
r = self.module.foo([[1, 2]])
|
||
|
|
assert r == [3]
|
||
|
|
|
||
|
|
r = self.module.foo([[1, 2], [3, 4]])
|
||
|
|
assert np.allclose(r, [3, 7])
|
||
|
|
|
||
|
|
r = self.module.foo([[1, 2], [3, 4], [5, 6]])
|
||
|
|
assert np.allclose(r, [3, 7, 11])
|
||
|
|
|
||
|
|
@pytest.mark.slow
|
||
|
|
def test_transpose(self):
|
||
|
|
r = self.module.trans([[]])
|
||
|
|
assert np.allclose(r.T, np.array([[]]))
|
||
|
|
|
||
|
|
r = self.module.trans([[1, 2]])
|
||
|
|
assert np.allclose(r, [[1.], [2.]])
|
||
|
|
|
||
|
|
r = self.module.trans([[1, 2, 3], [4, 5, 6]])
|
||
|
|
assert np.allclose(r, [[1, 4], [2, 5], [3, 6]])
|
||
|
|
|
||
|
|
@pytest.mark.slow
|
||
|
|
def test_flatten(self):
|
||
|
|
r = self.module.flatten([[]])
|
||
|
|
assert np.allclose(r, [])
|
||
|
|
|
||
|
|
r = self.module.flatten([[1, 2]])
|
||
|
|
assert np.allclose(r, [1, 2])
|
||
|
|
|
||
|
|
r = self.module.flatten([[1, 2, 3], [4, 5, 6]])
|
||
|
|
assert np.allclose(r, [1, 2, 3, 4, 5, 6])
|