110 lines
4.0 KiB
Python
110 lines
4.0 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.
|
|
|
|
# Standard Library Imports
|
|
|
|
# Third Party Imports
|
|
|
|
# Local Imports
|
|
from navigate.tools.common_dict_tools import update_stage_dict
|
|
|
|
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
|
|
def test_update_nested_dict():
|
|
from navigate.tools.common_dict_tools import update_nested_dict
|
|
|
|
test_dict = {"one": 2, "three": {4: {"metastasis": False}}}
|
|
|
|
test_dict_updated = update_nested_dict(
|
|
test_dict, lambda k, v: k == "metastasis", lambda x: True
|
|
)
|
|
|
|
assert test_dict_updated == {"one": 2, "three": {4: {"metastasis": True}}}
|
|
|
|
|
|
def create_mock_target():
|
|
"""Create a mock target object (Model or Controller) for testing."""
|
|
target = MagicMock()
|
|
target.configuration = {
|
|
"experiment": {
|
|
"MicroscopeState": {"channels": None},
|
|
"GalvoParameters": {},
|
|
"CameraParameters": {},
|
|
},
|
|
"etl_constants": {"ETLConstants": {"low": {}, "high": {}}},
|
|
}
|
|
return target
|
|
|
|
|
|
def create_mock_stage_target():
|
|
"""Create a mock target object (Model or Controller) for testing."""
|
|
target = MagicMock()
|
|
target.configuration = {
|
|
"experiment": {"StageParameters": {"x": None, "y": None, "z": None}}
|
|
}
|
|
return target
|
|
|
|
|
|
class UpdateStageDictTestCase(unittest.TestCase):
|
|
def test_update_single_axis(self):
|
|
target = create_mock_stage_target()
|
|
pos_dict = {"x_position": 10.0}
|
|
update_stage_dict(target, pos_dict)
|
|
self.assertEqual(
|
|
target.configuration["experiment"]["StageParameters"]["x"], 10.0
|
|
)
|
|
|
|
def test_update_multiple_axes(self):
|
|
target = create_mock_stage_target()
|
|
pos_dict = {"y_position": 5.0, "z_position": 2.5}
|
|
update_stage_dict(target, pos_dict)
|
|
self.assertEqual(
|
|
target.configuration["experiment"]["StageParameters"]["y"], 5.0
|
|
)
|
|
self.assertEqual(
|
|
target.configuration["experiment"]["StageParameters"]["z"], 2.5
|
|
)
|
|
|
|
def test_update_invalid_axis(self):
|
|
target = create_mock_stage_target()
|
|
pos_dict = {"invalid_axis": 3.14}
|
|
update_stage_dict(target, pos_dict)
|
|
self.assertNotIn(
|
|
"invalid_axis", target.configuration["experiment"]["StageParameters"]
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|