feat: init
This commit is contained in:
34
test/model/devices/daq/test_daq_base.py
Normal file
34
test/model/devices/daq/test_daq_base.py
Normal file
@@ -0,0 +1,34 @@
|
||||
from navigate.model.devices.daq.synthetic import SyntheticDAQ
|
||||
from test.model.dummy import DummyModel
|
||||
import numpy as np
|
||||
|
||||
|
||||
def test_initialize_daq():
|
||||
model = DummyModel()
|
||||
SyntheticDAQ(model.configuration)
|
||||
|
||||
|
||||
def test_calculate_all_waveforms():
|
||||
model = DummyModel()
|
||||
daq = SyntheticDAQ(model.configuration)
|
||||
microscope_state = model.configuration["experiment"]["MicroscopeState"]
|
||||
microscope_name = microscope_state["microscope_name"]
|
||||
exposure_times = {
|
||||
k: v["camera_exposure_time"] / 1000
|
||||
for k, v in microscope_state["channels"].items()
|
||||
}
|
||||
sweep_times = {
|
||||
k: 2 * v["camera_exposure_time"] / 1000
|
||||
for k, v in microscope_state["channels"].items()
|
||||
}
|
||||
waveform_dict = daq.calculate_all_waveforms(
|
||||
microscope_name, exposure_times, sweep_times
|
||||
)
|
||||
|
||||
for k, v in waveform_dict.items():
|
||||
channel = microscope_state["channels"][k]
|
||||
if not channel["is_selected"]:
|
||||
continue
|
||||
exposure_time = channel["camera_exposure_time"] / 1000
|
||||
print(k, channel["is_selected"], np.sum(v > 0), exposure_time)
|
||||
assert np.sum(v > 0) == daq.sample_rate * exposure_time
|
||||
78
test/model/devices/daq/test_daq_ni.py
Normal file
78
test/model/devices/daq/test_daq_ni.py
Normal file
@@ -0,0 +1,78 @@
|
||||
# Copyright (c) 2021-2025 The University of Texas Southwestern Medical Center.
|
||||
# All rights reserved.
|
||||
|
||||
# Redistribution and use in source and binary forms, with or without
|
||||
# modification, are permitted for academic and research use only (subject to the
|
||||
# limitations in the disclaimer below) provided that the following conditions are met:
|
||||
|
||||
# * Redistributions of source code must retain the above copyright notice,
|
||||
# this list of conditions and the following disclaimer.
|
||||
|
||||
# * Redistributions in binary form must reproduce the above copyright
|
||||
# notice, this list of conditions and the following disclaimer in the
|
||||
# documentation and/or other materials provided with the distribution.
|
||||
|
||||
# * Neither the name of the copyright holders nor the names of its
|
||||
# contributors may be used to endorse or promote products derived from this
|
||||
# software without specific prior written permission.
|
||||
|
||||
# NO EXPRESS OR IMPLIED LICENSES TO ANY PARTY'S PATENT RIGHTS ARE GRANTED BY
|
||||
# THIS LICENSE. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
|
||||
# CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
|
||||
# PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
|
||||
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
|
||||
# BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
|
||||
# IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
# POSSIBILITY OF SUCH DAMAGE.
|
||||
|
||||
# Standard Library Imports
|
||||
|
||||
# Third Party Imports
|
||||
import pytest
|
||||
|
||||
# Local Imports
|
||||
|
||||
|
||||
@pytest.mark.hardware
|
||||
def test_initialize_daq_ni():
|
||||
from navigate.model.devices.daq.ni import NIDAQ
|
||||
from test.model.dummy import DummyModel
|
||||
|
||||
model = DummyModel()
|
||||
daq = NIDAQ(model.configuration)
|
||||
daq.camera_trigger_task = None
|
||||
|
||||
|
||||
@pytest.mark.hardware
|
||||
def test_daq_ni_functions():
|
||||
from navigate.model.devices.daq.ni import NIDAQ
|
||||
from test.model.dummy import DummyModel
|
||||
|
||||
model = DummyModel()
|
||||
daq = NIDAQ(model.configuration)
|
||||
microscope_name = model.configuration["experiment"]["MicroscopeState"][
|
||||
"microscope_name"
|
||||
]
|
||||
|
||||
funcs = [
|
||||
"enable_microscope",
|
||||
"prepare_acquisition",
|
||||
"run_acquisition",
|
||||
"stop_acquisition",
|
||||
]
|
||||
args = [
|
||||
[microscope_name],
|
||||
[list(daq.waveform_dict.keys())[0]],
|
||||
None,
|
||||
None,
|
||||
]
|
||||
|
||||
for f, a in zip(funcs, args):
|
||||
if a is not None:
|
||||
getattr(daq, f)(*a)
|
||||
else:
|
||||
getattr(daq, f)()
|
||||
40
test/model/devices/daq/test_daq_synthetic.py
Normal file
40
test/model/devices/daq/test_daq_synthetic.py
Normal file
@@ -0,0 +1,40 @@
|
||||
def test_initialize_daq_synthetic():
|
||||
from navigate.model.devices.daq.synthetic import SyntheticDAQ
|
||||
from test.model.dummy import DummyModel
|
||||
|
||||
model = DummyModel()
|
||||
_ = SyntheticDAQ(model.configuration)
|
||||
|
||||
|
||||
def test_synthetic_daq_functions():
|
||||
import random
|
||||
|
||||
from navigate.model.devices.daq.synthetic import SyntheticDAQ
|
||||
from test.model.dummy import DummyModel
|
||||
|
||||
model = DummyModel()
|
||||
daq = SyntheticDAQ(model.configuration)
|
||||
microscope_name = model.configuration["experiment"]["MicroscopeState"][
|
||||
"microscope_name"
|
||||
]
|
||||
|
||||
funcs = [
|
||||
"add_camera",
|
||||
"prepare_acquisition",
|
||||
"run_acquisition",
|
||||
"stop_acquisition",
|
||||
"wait_acquisition_done",
|
||||
]
|
||||
args = [
|
||||
[microscope_name, model.camera[microscope_name]],
|
||||
[f"channel_{random.randint(1, 5)}"],
|
||||
None,
|
||||
None,
|
||||
None,
|
||||
]
|
||||
|
||||
for f, a in zip(funcs, args):
|
||||
if a is not None:
|
||||
getattr(daq, f)(*a)
|
||||
else:
|
||||
getattr(daq, f)()
|
||||
Reference in New Issue
Block a user