test: test different extractors

qiboteam · Jan 31, 2025 · 280b82e · 280b82e
1 parent 338f13b
commit 280b82e
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/tests/instruments/test_qrng.py b/tests/instruments/test_qrng.py
@@ -2,15 +2,20 @@
 import pytest
 from serial.serialutil import SerialException
 
-from qibolab.instruments.qrng import QRNG
+from qibolab.instruments.qrng import QRNG, ShaExtractor, ToeplitzExtractor
 
 RAW_BITS = 12
 """Number of bits in each QRNG sample."""
 
 
 @pytest.fixture
-def qrng(mocker):
-    qrng = QRNG(address="/dev/ttyACM0")
+def extractor():
+    return ShaExtractor()
+
+
+@pytest.fixture
+def qrng(mocker, extractor):
+    qrng = QRNG(address="/dev/ttyACM0", extractor=extractor)
     try:
         qrng.connect()
     except SerialException:
@@ -22,6 +27,7 @@ def read(n):
     return qrng
 
 
+@pytest.mark.parametrize("extractor", [ShaExtractor(), ToeplitzExtractor()])
 def test_qrng_random(qrng):
     data = qrng.random(1000)
     assert isinstance(data, np.ndarray)