final first commit

qibojit-benchmarks/benchmarks/tests/conftest.py

@@ -0,0 +1,33 @@
+NQUBITS = "3,4,5"
+MAX_QUBITS = "0,1,2,3,4"
+QIBO_BACKENDS = "qibojit,tensorflow,numpy"
+LIBRARIES = "qibo,qiskit,cirq,qsim,tfq,qulacs,projectq,hybridq"
+
+
+def pytest_addoption(parser):
+    parser.addoption("--nqubits", type=str, default=NQUBITS)
+    parser.addoption("--max-qubits", type=str, default=MAX_QUBITS)
+    parser.addoption("--qibo-backends", type=str, default=QIBO_BACKENDS)
+    parser.addoption("--libraries", type=str, default=LIBRARIES)
+    parser.addoption("--add", type=str, default="")
+
+
+def pytest_generate_tests(metafunc):
+    nqubits = [int(n) for n in metafunc.config.option.nqubits.split(",")]
+    library_options = [f"max_qubits={n}" for n in metafunc.config.option.max_qubits.split(",")]
+    backends = metafunc.config.option.qibo_backends.split(",")
+    libraries = metafunc.config.option.libraries.split(",")
+    additional = metafunc.config.option.add
+    if additional:
+        libraries.extend(additional.split(","))
+
+    if "nqubits" in metafunc.fixturenames:
+        metafunc.parametrize("nqubits", nqubits)
+    if "backend" in metafunc.fixturenames:
+        metafunc.parametrize("backend", backends)
+    if "library" in metafunc.fixturenames:
+        metafunc.parametrize("library", libraries)
+    if "library_options" in metafunc.fixturenames:
+        metafunc.parametrize("library_options", library_options)
+    if "transfer" in metafunc.fixturenames:
+        metafunc.parametrize("transfer", [False, True])