Files
navigate/test/model/features/test_feature_related_functions.py
2025-12-04 16:07:30 +08:00

134 lines
4.9 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
import pytest
# local imports
from navigate.model.features.feature_related_functions import (
convert_str_to_feature_list,
convert_feature_list_to_str,
)
from navigate.model.features.common_features import (
PrepareNextChannel,
LoopByCount,
ZStackAcquisition,
)
@pytest.mark.parametrize(
"feature_list_str, expected_list",
[
("", None),
("[]", []),
("[{'name': PrepareNextChannel}]", [{"name": PrepareNextChannel}]),
("[{'name': NonExistFeature}]", None),
(
"[({'name': PrepareNextChannel}, {'name': LoopByCount})]",
[({"name": PrepareNextChannel}, {"name": LoopByCount})],
),
(
"[({'name': PrepareNextChannel}, {'name': LoopByCount, 'args': (3,)})]",
[({"name": PrepareNextChannel}, {"name": LoopByCount, "args": (3,)})],
),
(
"[({'name': PrepareNextChannel}, {'name': LoopByCount, 'args': 3})]",
[({"name": PrepareNextChannel}, {"name": LoopByCount, "args": (3,)})],
),
(
"[({'name': PrepareNextChannel}, {'name': LoopByCount, 'args': (3)})]",
[({"name": PrepareNextChannel}, {"name": LoopByCount, "args": (3,)})],
),
(
"[(({'name': PrepareNextChannel}, {'name': LoopByCount, 'args': (3)}))]",
[({"name": PrepareNextChannel}, {"name": LoopByCount, "args": (3,)})],
),
(
"[{'name': ZStackAcquisition, 'args': (True, False, 'zstack',)}]",
[
{
"name": ZStackAcquisition,
"args": (
True,
False,
"zstack",
),
}
],
),
],
)
def test_convert_str_to_feature_list(feature_list_str, expected_list):
feature_list = convert_str_to_feature_list(feature_list_str)
assert feature_list == expected_list
@pytest.mark.parametrize(
"feature_list, expected_str",
[
(None, "[]"),
([], "[]"),
([{"name": PrepareNextChannel}], '[{"name": PrepareNextChannel,},]'),
(
[({"name": PrepareNextChannel}, {"name": LoopByCount})],
'[({"name": PrepareNextChannel,},{"name": LoopByCount,},),]',
),
(
[({"name": PrepareNextChannel}, {"name": LoopByCount, "args": (3,)})],
'[({"name": PrepareNextChannel,},{"name": LoopByCount,"args": (3,),},),]',
),
(
[
{
"name": ZStackAcquisition,
"args": (
True,
False,
"zstack",
),
}
],
'[{"name": ZStackAcquisition,"args": (True,False,"zstack",),},]',
),
],
)
def test_convert_feature_list_to_str(feature_list, expected_str):
feature_str = convert_feature_list_to_str(feature_list)
assert feature_str == expected_str
if feature_list:
assert convert_str_to_feature_list(feature_str) == feature_list