| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- import os
- import pytest
- import numpy as np
- from numpy.testing import assert_array_equal, assert_equal
- from . import util
- def get_docdir():
- # assuming that documentation tests are run from a source
- # directory
- return os.path.abspath(os.path.join(
- os.path.dirname(__file__),
- '..', '..', '..',
- 'doc', 'source', 'f2py', 'code'))
- pytestmark = pytest.mark.skipif(
- not os.path.isdir(get_docdir()),
- reason=('Could not find f2py documentation sources'
- f' ({get_docdir()} does not exists)'))
- def _path(*a):
- return os.path.join(*((get_docdir(),) + a))
- 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 = getattr(self.module, 'foo1')
- assert_equal(foo(), b'123456789A12')
- def test_asterisk2(self):
- foo = getattr(self.module, 'foo2')
- assert_equal(foo(2), b'12')
- assert_equal(foo(12), b'123456789A12')
- assert_equal(foo(24), 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
|
