46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
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)
|