Apply ruff

Author: mkitti

numcodecs/tests/test_pyzstd.py

@@ -13,38 +13,41 @@
     pyzstd = pytest.importorskip("pyzstd")
 
 test_data = [
-        b"Hello World!",
-        np.arange(113).tobytes(),
-        np.arange(10,15).tobytes(),
-        np.random.randint(3, 50, size=(53,)).tobytes()
+    b"Hello World!",
+    np.arange(113).tobytes(),
+    np.arange(10, 15).tobytes(),
+    np.random.randint(3, 50, size=(53,)).tobytes(),
 ]
 
+
 @pytest.mark.parametrize("input", test_data)
 def test_pyzstd_simple(input):
     z = Zstd()
     assert z.decode(pyzstd.compress(input)) == input
     assert pyzstd.decompress(z.encode(input)) == input
 
+
 @pytest.mark.xfail
 @pytest.mark.parametrize("input", test_data)
 def test_pyzstd_simple_multiple_frames(input):
     z = Zstd()
-    assert z.decode(pyzstd.compress(input)*2) == input*2
-    assert pyzstd.decompress(z.encode(input)*2) == input*2
+    assert z.decode(pyzstd.compress(input) * 2) == input * 2
+    assert pyzstd.decompress(z.encode(input) * 2) == input * 2
+
 
 @pytest.mark.parametrize("input", test_data)
 def test_pyzstd_streaming(input):
     pyzstd_c = pyzstd.ZstdCompressor()
     pyzstd_d = pyzstd.ZstdDecompressor()
     z = Zstd()
-    
+
     d_bytes = input
     pyzstd_c.compress(d_bytes)
     c_bytes = pyzstd_c.flush()
     assert z.decode(c_bytes) == d_bytes
     assert pyzstd_d.decompress(z.encode(d_bytes)) == d_bytes
 
     # Test multiple streaming frames
-    assert z.decode(c_bytes*2) == d_bytes*2
-    assert z.decode(c_bytes*3) == d_bytes*3
-    assert z.decode(c_bytes*99) == d_bytes*99
+    assert z.decode(c_bytes * 2) == d_bytes * 2
+    assert z.decode(c_bytes * 3) == d_bytes * 3
+    assert z.decode(c_bytes * 99) == d_bytes * 99

numcodecs/tests/test_zstd.py

@@ -131,7 +131,8 @@ def test_streaming_decompression():
     with pytest.raises(RuntimeError, match='Zstd decompression error: invalid input data'):
         codec.decode(bytes4)
 
-def generate_zstd_streaming_bytes(input: bytes, *, decompress: bool =False) -> bytes:
+
+def generate_zstd_streaming_bytes(input: bytes, *, decompress: bool = False) -> bytes:
     """
     Use the zstd command line interface to compress or decompress bytes in streaming mode.
     """
@@ -140,23 +141,21 @@ def generate_zstd_streaming_bytes(input: bytes, *, decompress: bool =False) -> b
     else:
         args = []
 
-    p = subprocess.run(
-            ["zstd", "--no-check", *args],
-            input=input,
-            capture_output=True
-    )
+    p = subprocess.run(["zstd", "--no-check", *args], input=input, capture_output=True)
     return p.stdout
 
+
 def view_zstd_streaming_bytes():
     bytes_val = generate_zstd_streaming_bytes(b"Hello world!")
     print(f"    bytes_val = {bytes_val}")
 
-    bytes3 = generate_zstd_streaming_bytes(b"ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz" * 1024 * 32)
+    bytes3 = generate_zstd_streaming_bytes(
+        b"ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz" * 1024 * 32
+    )
     print(f"    bytes3 = {bytes3}")
 
+
 def zstd_cli_available() -> bool:
     return not subprocess.run(
-        ["zstd", "-V"],
-        stdout=subprocess.DEVNULL,
-        stderr=subprocess.DEVNULL
+        ["zstd", "-V"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL
     ).returncode