Viewing file: test_size.py (1.14 KB) -rw-r--r-- Select action/file-type: (+) | (+) | (+) | Code (+) | Session (+) | (+) | SDB (+) | (+) | (+) | (+) | (+) | (+) |
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])
|