diff --git a/tests/instruments/test_qrng.py b/tests/instruments/test_qrng.py index 777188937..d345d1c97 100644 --- 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)