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
| import platform
|
| import pytest
|
| from numpy import array
|
| from . import util
|
| IS_S390X = platform.machine() == "s390x"
|
|
| @pytest.mark.slow
| class TestReturnCharacter(util.F2PyTest):
| def check_function(self, t, tname):
| if tname in ["t0", "t1", "s0", "s1"]:
| assert t("23") == b"2"
| r = t("ab")
| assert r == b"a"
| r = t(array("ab"))
| assert r == b"a"
| r = t(array(77, "u1"))
| assert r == b"M"
| elif tname in ["ts", "ss"]:
| assert t(23) == b"23"
| assert t("123456789abcdef") == b"123456789a"
| elif tname in ["t5", "s5"]:
| assert t(23) == b"23"
| assert t("ab") == b"ab"
| assert t("123456789abcdef") == b"12345"
| else:
| raise NotImplementedError
|
|
| class TestFReturnCharacter(TestReturnCharacter):
| sources = [
| util.getpath("tests", "src", "return_character", "foo77.f"),
| util.getpath("tests", "src", "return_character", "foo90.f90"),
| ]
|
| @pytest.mark.xfail(IS_S390X, reason="callback returns ' '")
| @pytest.mark.parametrize("name", ["t0", "t1", "t5", "s0", "s1", "s5", "ss"])
| def test_all_f77(self, name):
| self.check_function(getattr(self.module, name), name)
|
| @pytest.mark.xfail(IS_S390X, reason="callback returns ' '")
| @pytest.mark.parametrize("name", ["t0", "t1", "t5", "ts", "s0", "s1", "s5", "ss"])
| def test_all_f90(self, name):
| self.check_function(getattr(self.module.f90_return_char, name), name)
|
|
