test: move some code to lib/

1 files changed, 21 insertions, 0 deletions

diff --git a/test/py/lib/tests.py b/test/py/lib/tests.py
new file mode 100644
index 0000000..a676021
--- /dev/null
+++ b/test/py/lib/tests.py
@@ -0,0 +1,21 @@
+# Copyright (c) 2023 Egor Tensin <Egor.Tensin@gmail.com>
+# This file is part of the "cimple" project.
+# For details, see https://github.com/egor-tensin/cimple.
+# Distributed under the MIT License.
+
+import pytest
+
+
+# Reference: https://github.com/pytest-dev/pytest/issues/3628
+# Automatic generation of readable test IDs.
+def my_parametrize(names, values, ids=None, **kwargs):
+    _names = names.split(',') if isinstance(names, str) else names
+    if not ids:
+        if len(_names) == 1:
+            ids = [f'{names}={v}' for v in values]
+        else:
+            ids = [
+                '-'.join(f'{k}={v}' for k, v in zip(_names, combination))
+                for combination in values
+            ]
+    return pytest.mark.parametrize(names, values, ids=ids, **kwargs)