Fix zstd tests, coverage

Author: mkitti

numcodecs/tests/test_pyzstd.py

@@ -1,7 +1,9 @@
-import pytest
-import numpy as np
+# Check Zstd against pyzstd package
 from typing import TYPE_CHECKING
 
+import numpy as np
+import pytest
+
 try:
     from numcodecs.zstd import Zstd
 except ImportError:  # pragma: no cover
@@ -16,7 +18,7 @@
     b"Hello World!",
     np.arange(113).tobytes(),
     np.arange(10, 15).tobytes(),
-    np.random.randint(3, 50, size=(53,)).tobytes(),
+    np.random.randint(3, 50, size=(53,), dtype=np.uint16).tobytes(),
 ]
 
 
@@ -29,9 +31,14 @@ def test_pyzstd_simple(input):
 
 @pytest.mark.xfail
 @pytest.mark.parametrize("input", test_data)
-def test_pyzstd_simple_multiple_frames(input):
+def test_pyzstd_simple_multiple_frames_decode(input):
     z = Zstd()
     assert z.decode(pyzstd.compress(input) * 2) == input * 2
+
+
+@pytest.mark.parametrize("input", test_data)
+def test_pyzstd_simple_multiple_frames_encode(input):
+    z = Zstd()
     assert pyzstd.decompress(z.encode(input) * 2) == input * 2
 
 

numcodecs/tests/test_zstd.py

@@ -1,7 +1,7 @@
 import itertools
+import subprocess
 
 import numpy as np
-import subprocess
 import pytest
 
 try:
@@ -99,6 +99,8 @@ def test_streaming_decompression():
 
     # If the zstd command line interface is available, check the bytes
     cli = zstd_cli_available()
+    if cli:
+        view_zstd_streaming_bytes()
 
     # Encode bytes directly that were the result of streaming compression
     bytes_val = b'(\xb5/\xfd\x00Xa\x00\x00Hello World!'