summaryrefslogtreecommitdiff
path: root/numpy/f2py/tests/test_kind.py
blob: 78a11fc6c403f9c54d2bcfbebcb93eae0c3f4f54 (plain)
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
import os
import pytest

from numpy.testing import assert_
from numpy.f2py.crackfortran import (
    _selected_int_kind_func as selected_int_kind,
    _selected_real_kind_func as selected_real_kind,
)
from . import util


class TestKind(util.F2PyTest):
    sources = [util.getpath("tests", "src", "kind", "foo.f90")]

    @pytest.mark.slow
    def test_all(self):
        selectedrealkind = self.module.selectedrealkind
        selectedintkind = self.module.selectedintkind

        for i in range(40):
            assert_(
                selectedintkind(i) in [selected_int_kind(i), -1],
                "selectedintkind(%s): expected %r but got %r" %
                (i, selected_int_kind(i), selectedintkind(i)),
            )

        for i in range(20):
            assert_(
                selectedrealkind(i) in [selected_real_kind(i), -1],
                "selectedrealkind(%s): expected %r but got %r" %
                (i, selected_real_kind(i), selectedrealkind(i)),
            )