Convert test_codecs.py to pytest

Now used as test-command for cibuildwheel

Author: flaviomartins

python_bindings/MANIFEST.in

@@ -2,6 +2,7 @@ include pyfastpfor.cc
 include pyproject.toml
 recursive-include ./fastpfor/src *.cpp *.c
 recursive-include ./fastpfor/headers *.h
+recursive-include tests *.py
 include requirements.txt
 
 global-exclude fastpfor/src/benchbitpacking.cpp

python_bindings/pyproject.toml

@@ -51,8 +51,9 @@ build-frontend = "build"
 # Published wheels must run on any CPU of the target architecture, so build with
 # a portable SIMD baseline rather than -march=native (see setup.py:simd_flags).
 environment = { PYFASTPFOR_PORTABLE = "1" }
-# Smoke-test every built wheel: import it and exercise the codec factory.
-test-command = 'python -c "import pyfastpfor; assert pyfastpfor.getCodecList(); print(pyfastpfor.__version__, len(pyfastpfor.getCodecList()), \"codecs\")"'
+# Test every built wheel: round-trip every codec (see tests/test_codecs.py).
+test-requires = "pytest"
+test-command = "pytest {package}/tests -v"
 
 [tool.cibuildwheel.linux]
 archs = ["native"]

python_bindings/tests/test_codecs.py

@@ -2,12 +2,22 @@
 
 from pyfastpfor import *
 import numpy as np
+import pytest
 import random
 import time
 random.seed(0)
 
+# Be careful changing these numbers: If the array size
+# is too large there will be an integer overflow.
+ARRAY_SIZES = [1, 8, 64, 1024, 1024 * 1024 * 4]
+MAX_VALUES = [256, 512, 2048]
+
+
+@pytest.mark.parametrize("maxVal", MAX_VALUES)
+@pytest.mark.parametrize("arrSize", ARRAY_SIZES)
+def test_codecs_roundtrip(arrSize, maxVal):
+    codecList = getCodecList()
 
-def oneTest(arrSize, maxVal, codecList):
     inpSmall = np.array(np.random.randint(
         0, maxVal, arrSize), dtype=np.uint32).ravel()
     inpPrefSum = np.array(inpSmall, dtype=np.uint32, copy=True).ravel()
@@ -38,9 +48,9 @@ def oneTest(arrSize, maxVal, codecList):
         for diffType in range(3):
 
             if diffType == 0:
-                inp = np.array([e for e in inpSmall], dtype=np.uint32).ravel()
+                inp = np.array(inpSmall, dtype=np.uint32, copy=True).ravel()
             else:
-                inp = np.array([e for e in inpPrefSum], dtype=np.uint32).ravel()
+                inp = np.array(inpPrefSum, dtype=np.uint32, copy=True).ravel()
 
             if diffType == 1:
                 delta1(inp, arrSize)
@@ -77,20 +87,4 @@ def oneTest(arrSize, maxVal, codecList):
 
 
 if __name__ == '__main__':
-
-    codecList = getCodecList()
-
-    # Be careful changing these numbers: If the array size
-    # is too large there will be an integer overflow.
-    arraySizes = [1, 8, 64, 1024, 1024 * 1024 * 4]
-    maxValues = [256, 512, 2048]
-
-    i = 0
-    for arraySize in arraySizes:
-        for maxValue in maxValues:
-            print("--------------------------------------------------------------------------------")
-            print("                              Test Case %d/%d" % (i, len(arraySizes)*len(maxValues)))
-            print("--------------------------------------------------------------------------------")
-            oneTest(arraySize, maxValue, codecList)
-            print("\n")
-            i += 1
+    raise SystemExit(pytest.main([__file__, "-v"]))