summaryrefslogtreecommitdiff
path: root/numpy/f2py/tests/test_string.py
diff options
context:
space:
mode:
authorRohit Goswami <rog32@hi.is>2021-12-02 21:21:34 +0000
committerRohit Goswami <rog32@hi.is>2021-12-05 22:23:55 +0000
commite4ca1fbcc96f956708f2deacb6ae1806cd132fe5 (patch)
tree6001fb86ee2e1982d712977fad8a5c7dc7a6bc89 /numpy/f2py/tests/test_string.py
parentcd24ca1ef578924bad8c5c31f8939e64e353c256 (diff)
downloadnumpy-e4ca1fbcc96f956708f2deacb6ae1806cd132fe5.tar.gz
MAINT: Rework whitespace to pass PR checks
Co-authored-by: Charles Harris <charlesr.harris@gmail.com>
Diffstat (limited to 'numpy/f2py/tests/test_string.py')
-rw-r--r--numpy/f2py/tests/test_string.py80
1 files changed, 40 insertions, 40 deletions
diff --git a/numpy/f2py/tests/test_string.py b/numpy/f2py/tests/test_string.py
index b5d18ce4b..1a6d59610 100644
--- a/numpy/f2py/tests/test_string.py
+++ b/numpy/f2py/tests/test_string.py
@@ -7,21 +7,21 @@ from . import util
class TestString(util.F2PyTest):
- sources = [util.getpath('tests', 'src', 'string', 'char.f90')]
+ sources = [util.getpath("tests", "src", "string", "char.f90")]
@pytest.mark.slow
def test_char(self):
- strings = np.array(['ab', 'cd', 'ef'], dtype='c').T
- inp, out = self.module.char_test.change_strings(strings,
- strings.shape[1])
+ strings = np.array(["ab", "cd", "ef"], dtype="c").T
+ inp, out = self.module.char_test.change_strings(
+ strings, strings.shape[1])
assert_array_equal(inp, strings)
expected = strings.copy()
- expected[1, :] = 'AAA'
+ expected[1, :] = "AAA"
assert_array_equal(out, expected)
class TestDocStringArguments(util.F2PyTest):
- suffix = '.f'
+ suffix = ".f"
code = """
C FILE: STRING.F
@@ -39,21 +39,21 @@ C END OF FILE STRING.F
"""
def test_example(self):
- a = np.array(b'123\0\0')
- b = np.array(b'123\0\0')
- c = np.array(b'123')
- d = np.array(b'123')
+ a = np.array(b"123\0\0")
+ b = np.array(b"123\0\0")
+ c = np.array(b"123")
+ d = np.array(b"123")
self.module.foo(a, b, c, d)
- assert a.tobytes() == b'123\0\0'
- assert b.tobytes() == b'B23\0\0'
- assert c.tobytes() == b'123'
- assert d.tobytes() == b'D23'
+ assert a.tobytes() == b"123\0\0"
+ assert b.tobytes() == b"B23\0\0"
+ assert c.tobytes() == b"123"
+ assert d.tobytes() == b"D23"
class TestFixedString(util.F2PyTest):
- suffix = '.f90'
+ suffix = ".f90"
code = textwrap.dedent("""
function sint(s) result(i)
@@ -109,41 +109,41 @@ class TestFixedString(util.F2PyTest):
end = len(s)
i = 0
for j in range(start, min(end, len(s))):
- i += s[j] * 10 ** j
+ i += s[j] * 10**j
return i
- def _get_input(self, intent='in'):
- if intent in ['in']:
- yield ''
- yield '1'
- yield '1234'
- yield '12345'
- yield b''
- yield b'\0'
- yield b'1'
- yield b'\01'
- yield b'1\0'
- yield b'1234'
- yield b'12345'
- yield np.ndarray((), np.bytes_, buffer=b'') # array(b'', dtype='|S0')
- yield np.array(b'') # array(b'', dtype='|S1')
- yield np.array(b'\0')
- yield np.array(b'1')
- yield np.array(b'1\0')
- yield np.array(b'\01')
- yield np.array(b'1234')
- yield np.array(b'123\0')
- yield np.array(b'12345')
+ def _get_input(self, intent="in"):
+ if intent in ["in"]:
+ yield ""
+ yield "1"
+ yield "1234"
+ yield "12345"
+ yield b""
+ yield b"\0"
+ yield b"1"
+ yield b"\01"
+ yield b"1\0"
+ yield b"1234"
+ yield b"12345"
+ yield np.ndarray((), np.bytes_, buffer=b"") # array(b'', dtype='|S0')
+ yield np.array(b"") # array(b'', dtype='|S1')
+ yield np.array(b"\0")
+ yield np.array(b"1")
+ yield np.array(b"1\0")
+ yield np.array(b"\01")
+ yield np.array(b"1234")
+ yield np.array(b"123\0")
+ yield np.array(b"12345")
def test_intent_in(self):
for s in self._get_input():
r = self.module.test_in_bytes4(s)
# also checks that s is not changed inplace
expected = self._sint(s, end=4)
- assert r == expected, (s)
+ assert r == expected, s
def test_intent_inout(self):
- for s in self._get_input(intent='inout'):
+ for s in self._get_input(intent="inout"):
rest = self._sint(s, start=4)
r = self.module.test_inout_bytes4(s)
expected = self._sint(s, end=4)