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
| import pytest
|
| from numpy import array
| from . import util
| import platform
|
| IS_S390X = platform.machine() == "s390x"
|
|
| 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".split(","))
| 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".split(","))
| def test_all_f90(self, name):
| self.check_function(getattr(self.module.f90_return_char, name), name)
|
|
