134 lines
5.1 KiB
Python
134 lines
5.1 KiB
Python
# 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.
|
|
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
from navigate.model.devices.galvo.ni import NIGalvo
|
|
from navigate.config import (
|
|
load_configs,
|
|
get_configuration_paths,
|
|
verify_configuration,
|
|
verify_waveform_constants,
|
|
)
|
|
from multiprocessing import Manager
|
|
|
|
|
|
class TestNIGalvo(unittest.TestCase):
|
|
"""Unit tests for the Galvo NI Device."""
|
|
|
|
def setUp(self) -> None:
|
|
"""Set up the configuration, experiment, etc."""
|
|
self.manager = Manager()
|
|
self.parent_dict = {}
|
|
|
|
(
|
|
configuration_path,
|
|
experiment_path,
|
|
waveform_constants_path,
|
|
rest_api_path,
|
|
waveform_templates_path,
|
|
gui_configuration_path,
|
|
multi_positions_path,
|
|
) = get_configuration_paths()
|
|
|
|
self.configuration = load_configs(
|
|
self.manager,
|
|
configuration=configuration_path,
|
|
experiment=experiment_path,
|
|
waveform_constants=waveform_constants_path,
|
|
rest_api_config=rest_api_path,
|
|
waveform_templates=waveform_templates_path,
|
|
gui_configuration_path=gui_configuration_path,
|
|
)
|
|
verify_configuration(self.manager, self.configuration)
|
|
verify_waveform_constants(self.manager, self.configuration)
|
|
self.microscope_name = "Mesoscale"
|
|
self.device_connection = MagicMock()
|
|
galvo_id = 0
|
|
|
|
self.galvo = NIGalvo(
|
|
microscope_name=self.microscope_name,
|
|
device_connection=self.device_connection,
|
|
configuration=self.configuration,
|
|
device_id=galvo_id,
|
|
)
|
|
|
|
def tearDown(self):
|
|
"""Tear down the multiprocessing manager."""
|
|
self.manager.shutdown()
|
|
|
|
def test_galvo_ni_initialization(self):
|
|
# Parent Class Super Init
|
|
assert self.galvo.microscope_name == "Mesoscale"
|
|
assert self.galvo.galvo_name == "Galvo 0"
|
|
assert self.galvo.sample_rate == 100000
|
|
assert (
|
|
self.galvo.camera_delay
|
|
== self.configuration["configuration"]["microscopes"][self.microscope_name][
|
|
"camera"
|
|
]["delay"]
|
|
/ 1000
|
|
)
|
|
assert self.galvo.galvo_max_voltage == 5
|
|
assert self.galvo.galvo_min_voltage == -5
|
|
assert self.galvo.galvo_waveform == "sawtooth" or "sine"
|
|
assert self.galvo.waveform_dict == {}
|
|
|
|
# NIGalvo Init
|
|
assert self.galvo.trigger_source == "/PXI6259/PFI0"
|
|
assert hasattr(self.galvo, "daq")
|
|
|
|
def test_adjust(self):
|
|
sweep_times = {"channel_1": 0.3, "channel_2": 0.4, "channel_3": 0.5}
|
|
|
|
exposure_times = {"channel_1": 0.25, "channel_2": 0.35, "channel_3": 0.45}
|
|
|
|
waveforms = self.galvo.adjust(
|
|
exposure_times=exposure_times, sweep_times=sweep_times
|
|
)
|
|
|
|
assert type(waveforms) == dict
|
|
self.device_connection.assert_not_called()
|
|
|
|
for channel_key, channel_setting in self.configuration["experiment"][
|
|
"MicroscopeState"
|
|
]["channels"].items():
|
|
if channel_setting["is_selected"]:
|
|
assert channel_key in waveforms.keys()
|
|
|
|
self.device_connection.analog_outputs.__setitem__.assert_called_with(
|
|
self.galvo.device_config["hardware"]["channel"],
|
|
{
|
|
"trigger_source": self.galvo.trigger_source,
|
|
"waveform": waveforms,
|
|
},
|
|
)
|