test_assumed_shape.py 949 B

123456789101112131415161718192021222324252627282930313233
  1. from __future__ import division, absolute_import, print_function
  2. import os
  3. import pytest
  4. from numpy.testing import assert_
  5. from . import util
  6. def _path(*a):
  7. return os.path.join(*((os.path.dirname(__file__),) + a))
  8. class TestAssumedShapeSumExample(util.F2PyTest):
  9. sources = [_path('src', 'assumed_shape', 'foo_free.f90'),
  10. _path('src', 'assumed_shape', 'foo_use.f90'),
  11. _path('src', 'assumed_shape', 'precision.f90'),
  12. _path('src', 'assumed_shape', 'foo_mod.f90'),
  13. ]
  14. @pytest.mark.slow
  15. def test_all(self):
  16. r = self.module.fsum([1, 2])
  17. assert_(r == 3, repr(r))
  18. r = self.module.sum([1, 2])
  19. assert_(r == 3, repr(r))
  20. r = self.module.sum_with_use([1, 2])
  21. assert_(r == 3, repr(r))
  22. r = self.module.mod.sum([1, 2])
  23. assert_(r == 3, repr(r))
  24. r = self.module.mod.fsum([1, 2])
  25. assert_(r == 3, repr(r))