test_kind.py 1012 B

1234567891011121314151617181920212223242526272829303132
  1. import os
  2. import pytest
  3. from numpy.testing import assert_
  4. from numpy.f2py.crackfortran import (
  5. _selected_int_kind_func as selected_int_kind,
  6. _selected_real_kind_func as selected_real_kind
  7. )
  8. from . import util
  9. def _path(*a):
  10. return os.path.join(*((os.path.dirname(__file__),) + a))
  11. class TestKind(util.F2PyTest):
  12. sources = [_path('src', 'kind', 'foo.f90')]
  13. @pytest.mark.slow
  14. def test_all(self):
  15. selectedrealkind = self.module.selectedrealkind
  16. selectedintkind = self.module.selectedintkind
  17. for i in range(40):
  18. assert_(selectedintkind(i) in [selected_int_kind(i), -1],
  19. 'selectedintkind(%s): expected %r but got %r' %
  20. (i, selected_int_kind(i), selectedintkind(i)))
  21. for i in range(20):
  22. assert_(selectedrealkind(i) in [selected_real_kind(i), -1],
  23. 'selectedrealkind(%s): expected %r but got %r' %
  24. (i, selected_real_kind(i), selectedrealkind(i)))