File size: 1,850 Bytes
7885a28 |
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 |
import pytest
import numpy as np
from numpy.testing import assert_array_equal, assert_equal
from . import util
from pathlib import Path
def get_docdir():
parents = Path(__file__).resolve().parents
try:
# Assumes that spin is used to run tests
nproot = parents[8]
except IndexError:
docdir = None
else:
docdir = nproot / "doc" / "source" / "f2py" / "code"
if docdir and docdir.is_dir():
return docdir
# Assumes that an editable install is used to run tests
return parents[3] / "doc" / "source" / "f2py" / "code"
pytestmark = pytest.mark.skipif(
not get_docdir().is_dir(),
reason=f"Could not find f2py documentation sources"
f"({get_docdir()} does not exist)",
)
def _path(*args):
return get_docdir().joinpath(*args)
@pytest.mark.slow
class TestDocAdvanced(util.F2PyTest):
# options = ['--debug-capi', '--build-dir', '/tmp/build-f2py']
sources = [_path('asterisk1.f90'), _path('asterisk2.f90'),
_path('ftype.f')]
def test_asterisk1(self):
foo = self.module.foo1
assert_equal(foo(), b'123456789A12')
def test_asterisk2(self):
foo = self.module.foo2
assert_equal(foo(2), b'12')
assert_equal(foo(12), b'123456789A12')
assert_equal(foo(20), b'123456789A123456789B')
def test_ftype(self):
ftype = self.module
ftype.foo()
assert_equal(ftype.data.a, 0)
ftype.data.a = 3
ftype.data.x = [1, 2, 3]
assert_equal(ftype.data.a, 3)
assert_array_equal(ftype.data.x,
np.array([1, 2, 3], dtype=np.float32))
ftype.data.x[1] = 45
assert_array_equal(ftype.data.x,
np.array([1, 45, 3], dtype=np.float32))
# TODO: implement test methods for other example Fortran codes
|