Merge branch 'feature/pytest' into develop

dd1ea3b5 · Federico Vaga · a86c1548 · 038c9976 · dd1ea3b5 · dd1ea3b5
Commit dd1ea3b5 authored Aug 18, 2020 by Federico Vaga
12 changed files
--- a/PyAdcLib/PyAdcLib/PyAdcGenericFake.py
+++ b/PyAdcLib/PyAdcLib/PyAdcGenericFake.py
 from PyAdcLib import ADC_Generic

 class AdcGenericFake(ADC_Generic):
-    BOARD_NAME = b"adc-genericfake"
+    BOARD_NAME = "adc-genericfake"

    def __init__(self, devid):
        super(AdcGenericFake, self).__init__(devid)
--- a/PyAdcLib/PyAdcLib/PyAdcLib.py
+++ b/PyAdcLib/PyAdcLib/PyAdcLib.py
--- a/PyAdcLib/PyAdcLib/PyAdcLibFmcAdc100m14b4ch.py
+++ b/PyAdcLib/PyAdcLib/PyAdcLibFmcAdc100m14b4ch.py
-from .PyAdcLib import ADC_Generic
-
-class FmcAdc100m14b4ch(ADC_Generic):
-    BOARD_NAME = b"fmc-adc-100m14b4cha"
-
-    ADC_CONF_100M14B4CHA_CHN_RANGE_N = 3
-
-    ADC_CONF_100M14B4CHA_CHN_RANGE_OPEN_DRAIN = 0
-    ADC_CONF_100M14B4CHA_CHN_RANGE_100mV = 0x23
-    ADC_CONF_100M14B4CHA_CHN_RANGE_1V = 0x11
-    ADC_CONF_100M14B4CHA_CHN_RANGE_10V = 0x45
-    ADC_CONF_100M14B4CHA_CHN_RANGE_100mV_CAL = 0x42
-    ADC_CONF_100M14B4CHA_CHN_RANGE_1V_CAL = 0x40
-    ADC_CONF_100M14B4CHA_CHN_RANGE_10V_CAL = 0x44
-
-    ADC_CONF_100M14B4CHA_BUF_KMALLOC = 0
-    ADC_CONF_100M14B4CHA_BUF_VMALLOC = 1
-
-    ADC_CONF_100M14B4CHA_BUF_TYPE = 0
-    ADC_CONF_100M14B4CHA_TRG_SW_EN = 1
-    ADC_CONF_100M14B4CHA_ACQ_MSHOT_MAX = 2
-    ADC_CONF_100M14B4CHA_BUF_SIZE_KB = 3
-    ADC_CONF_100M14B4CHA_TRG_ALT_EN = 4
-    __ADC_CONF_100M14B4CHA_LAST_INDEX = 5
-
-    def __init__(self, devid):
-        super(FmcAdc100m14b4ch, self).__init__(devid)
--- a/PyAdcLib/PyAdcLib/PyAdcZioFake.py
+++ b/PyAdcLib/PyAdcLib/PyAdcZioFake.py
 from PyAdcLib import ADC_Generic

 class AdcZioFake(ADC_Generic):
-    BOARD_NAME = b"adc-ziofake"
+    BOARD_NAME = "adc-ziofake"

    def __init__(self, devid):
        super(AdcZioFake, self).__init__(devid)
--- a/PyAdcLib/PyAdcLib/__init__.py
+++ b/PyAdcLib/PyAdcLib/__init__.py
@@ -7,5 +7,13 @@
 SPDX-License-Identifier: LGPL-3.0-or-later
 """

+from .PyAdcLib import PyAdcConf, PyAdcAbstract, PyFmcAdc100m14b4ch, timeval, \
+     adc_buffer_get_sample
+
 __all__ = (
+    "PyAdcConf",
+    "PyAdcAbstract",
+    "PyFmcAdc100m14b4ch",
+    "timeval",
+    "adc_buffer_get_sample"
 )
--- a/lib/Makefile
+++ b/lib/Makefile
@@ -6,7 +6,8 @@
 -include Makefile.specific

 # include parent_common.mk for buildsystem's defines
-REPO_PARENT ?= ../..
+IGNORE_CPU_SUFFIX := y
+REPO_PARENT ?=
 -include $(REPO_PARENT)/parent_common.mk

 ZIO ?= $(HOME)/git/zio

--- a/pytest/conftest.py
+++ b/pytest/conftest.py
+"""
+SPDX-License-Identifier: GPL-3.0-or-later
+SPDX-FileCopyrightText: 2020 CERN
+"""
+
+import pytest
+from PyAdcLib import *
+
+@pytest.fixture(scope="function")
+def adc100m14b4cha():
+    adc = PyFmcAdc100m14b4ch(pytest.adc_id)
+    yield adc
+    adc.acq_stop(0)
+
+@pytest.fixture(scope="function")
+def adc100m14b4cha_def_cfg():
+    adc = PyFmcAdc100m14b4ch(pytest.adc_id)
+    for ch in range(4):
+        conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_CHN, ch)
+        conf.value_set(PyAdcConf.ADC_CONF_CHN_RANGE, 0x45)
+        conf.value_set(PyAdcConf.ADC_CONF_CHN_TERMINATION, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_CHN_OFFSET, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_CHN_SATURATION, 0x7FFF)
+        adc.apply_config(conf, 0)
+
+    for ch in range(4):
+        conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_TRG_THR, ch)
+        conf.value_set(PyAdcConf.ADC_CONF_TRG_THR_ENABLE, 0)
+        adc.apply_config(conf, 0)
+
+    conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_TRG_EXT, 0)
+    conf.value_set(PyAdcConf.ADC_CONF_TRG_EXT_ENABLE, 0)
+    adc.apply_config(conf, 0)
+
+    yield adc
+    adc.acq_stop(0)
+
+@pytest.fixture(scope="function")
+def adc_simple():
+    adc = PyFmcAdc100m14b4ch(pytest.adc_id)
+    for ch in range(4):
+        conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_CHN, ch)
+        conf.value_set(PyAdcConf.ADC_CONF_CHN_RANGE, 0x45)
+        conf.value_set(PyAdcConf.ADC_CONF_CHN_TERMINATION, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_CHN_OFFSET, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_CHN_SATURATION, 0x7FFF)
+        adc.apply_config(conf, 0)
+
+    for ch in range(4):
+        conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_TRG_THR, ch)
+        conf.value_set(PyAdcConf.ADC_CONF_TRG_THR_ENABLE, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_TRG_THR_POLARITY, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_TRG_THR_DELAY, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_TRG_THR_THRESHOLD, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_TRG_THR_HYSTERESIS, 0)
+        adc.apply_config(conf, 0)
+
+    conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_TRG_EXT, 0)
+    conf.value_set(PyAdcConf.ADC_CONF_TRG_EXT_ENABLE, 0)
+    adc.apply_config(conf, 0)
+
+    conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_TRG_TIM, 0)
+    conf.value_set(PyAdcConf.ADC_CONF_TRG_TIM_ENABLE, 0)
+    conf.value_set(PyAdcConf.ADC_CONF_TRG_TIM_SECONDS, 0)
+    conf.value_set(PyAdcConf.ADC_CONF_TRG_TIM_NANO_SECONDS, 0)
+    adc.apply_config(conf, 0)
+
+    conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_ACQ, 0)
+    conf.value_set(PyAdcConf.ADC_CONF_ACQ_PRE_SAMP, 1)
+    conf.value_set(PyAdcConf.ADC_CONF_ACQ_POST_SAMP, 2)
+    conf.value_set(PyAdcConf.ADC_CONF_ACQ_N_SHOTS, 1)
+    adc.apply_config(conf, 0)
+
+    yield adc
+    adc.acq_stop(0)
+
+def pytest_addoption(parser):
+    parser.addoption("--type", choices=[PyFmcAdc100m14b4ch.BOARD_NAME],
+                     required=True, help="Fmc ADC type")
+    parser.addoption("--id", type=lambda x : int(x, 16),
+                     required=True, help="Fmc ADC Identifier")
+
+def pytest_configure(config):
+    pytest.adc_type = config.getoption("--type")
+    pytest.adc_id = config.getoption("--id")
--- a/pytest/fmc-adc-100m14b4cha/test_adc_acquisition.py
+++ b/pytest/fmc-adc-100m14b4cha/test_adc_acquisition.py
+"""
+SPDX-License-Identifier: GPL-3.0-or-later
+SPDX-FileCopyrightText: 2020 CERN
+"""
+
+import pytest
+import time
+from PyAdcLib import PyAdcConf, PyFmcAdc100m14b4ch, timeval, \
+     adc_buffer_get_sample
+
+@pytest.fixture(scope="function", params=[1 << i for i in range(14)])
+def adc_simple_pattern(adc_simple, request):
+    adc_simple.pattern_data = request.param
+    yield adc_simple
+    adc_simple.pattern_data = None
+
+class TestAdcAcquisitionPattern(object):
+
+    def test_adc_acq_poll_fail(self, adc_simple):
+        """Test that the poll fails with timeout if there are no triggers"""
+        conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_ACQ, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_ACQ_N_SHOTS, 1)
+        adc_simple.apply_config(conf, 0)
+
+        timeout = timeval(0,0)
+        adc_simple.acq_start(PyFmcAdc100m14b4ch.ADC_F_FLUSH,
+                                         timeout)
+        with pytest.raises(OSError):
+            timeout = timeval(2,0)
+            adc_simple.acq_poll(0, timeout)
+
+    @pytest.mark.parametrize("nshots", range(1, 10))
+    def test_adc_acq_poll(self, adc_simple, nshots):
+        conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_ACQ, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_ACQ_N_SHOTS, nshots)
+        adc_simple.apply_config(conf, 0)
+
+        timeout = timeval(0,0)
+        adc_simple.acq_start(PyFmcAdc100m14b4ch.ADC_F_FLUSH,
+                                         timeout)
+        for i in range(nshots):
+            adc_simple.trigger_fire()
+            time.sleep(0.1)
+        timeout = timeval(10,0)
+        adc_simple.acq_poll(0, timeout)
+
+    @pytest.mark.parametrize("pre_samples,post_samples",
+                             [(10**x, 10**y) for x in range(1, 3) for y in range(1, 3)])
+    @pytest.mark.parametrize("nshots", range(1, 10))
+    def test_adc_acq_basic(self, adc_simple_pattern,
+                           pre_samples, post_samples, nshots):
+        """Test that we can get"""
+        conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_ACQ, 0)
+        conf.value_set(PyAdcConf.ADC_CONF_ACQ_PRE_SAMP, pre_samples)
+        conf.value_set(PyAdcConf.ADC_CONF_ACQ_POST_SAMP, post_samples)
+        conf.value_set(PyAdcConf.ADC_CONF_ACQ_N_SHOTS, nshots)
+        adc_simple_pattern.apply_config(conf, 0)
+
+        nsamples = pre_samples + post_samples
+        buf = adc_simple_pattern.request_buffer(nsamples * nshots, None, 0)
+        adc_simple_pattern.acq_start(PyFmcAdc100m14b4ch.ADC_F_FLUSH,
+                                         timeval(0, 0))
+        for n in range(nshots):
+            time.sleep(1e-8 * pre_samples)
+            adc_simple_pattern.trigger_fire()
+        adc_simple_pattern.acq_poll(0, timeval(10, 0))
+        adc_simple_pattern.fill_buffer(buf, 0, timeval(10, 0))
+        assert buf.contents.nsamples == nsamples
+
+        pattern = adc_simple_pattern.pattern_data
+        for n in range(nshots):
+            for chan in range(4):
+                for s in range(nsamples):
+                    assert adc_buffer_get_sample(buf, chan, s) >> 2 == pattern, \
+                      buf.contents.data[:nsamples]
+
+        adc_simple_pattern.release_buffer(buf, None)
--- a/pytest/fmc-adc-100m14b4cha/test_adc_getter_setter.py
+++ b/pytest/fmc-adc-100m14b4cha/test_adc_getter_setter.py
--- a/pytest/fmc-adc-100m14b4cha/test_adc_time.py
+++ b/pytest/fmc-adc-100m14b4cha/test_adc_time.py
+"""
+SPDX-License-Identifier: GPL-3.0-or-later
+SPDX-FileCopyrightText: 2020 CERN
+"""
+
+import pytest
+import time
+from PyAdcLib import PyAdcConf, PyFmcAdc100m14b4ch
+
+class TestAdcTime(object):
+
+    def test_time_flows(self, adc100m14b4cha):
+        """Just check that the time flows more or less correctly second by
+        second for a minute"""
+        conf = PyAdcConf(PyAdcConf.ADC_CONF_TYPE_BRD, 0)
+        conf.mask_set(PyAdcConf.ADC_CONF_UTC_TIMING_BASE_S)
+        adc100m14b4cha.retrieve_config(conf)
+        t_prev = conf.value_get(PyAdcConf.ADC_CONF_UTC_TIMING_BASE_S)
+        for i in range(20):
+            time.sleep(1)
+            adc100m14b4cha.retrieve_config(conf)
+            sec = conf.value_get(PyAdcConf.ADC_CONF_UTC_TIMING_BASE_S)
+            assert t_prev + 1 == sec
+            t_prev = sec
--- a/pytest/fmc-adc-100m14b4cha/test_adc_trigger.py
+++ b/pytest/fmc-adc-100m14b4cha/test_adc_trigger.py
--- a/pytest/pytest.ini
+++ b/pytest/pytest.ini
+# SPDX-License-Identifier: GPL-3.0-or-later
+# SPDX-FileCopyrightText: 2020 CERN
+
+[pytest]
+addopts = -v -p no:cacheprovider
\ No newline at end of file