203 lines
7.4 KiB
Python
203 lines
7.4 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
|
|
import numpy as np
|
|
|
|
# Local application imports
|
|
from navigate.controller.sub_controllers import AutofocusPopupController
|
|
from navigate.view.popups.autofocus_setting_popup import AutofocusPopup
|
|
|
|
|
|
class TestAutofocusPopupController:
|
|
"""Class for testing autofocus popup controller
|
|
|
|
Methods
|
|
-------
|
|
test_init()
|
|
Tests that the controller is initialized correctly
|
|
test_attr()
|
|
Tests that the attributes are initialized correctly
|
|
test_populate_experiment_values()
|
|
Tests that the values are populated correctly
|
|
test_update_experiment_values()
|
|
Tests that the values are updated correctly
|
|
test_start_autofocus()
|
|
Tests that the start autofocus function works correctly
|
|
test_display_plot()
|
|
Tests that the display plot function works correctly
|
|
"""
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup_class(self, dummy_controller):
|
|
"""Setup for testing autofocus popup controller
|
|
|
|
Parameters
|
|
----------
|
|
dummy_controller : DummyController
|
|
Dummy controller for testing
|
|
|
|
"""
|
|
autofocus_popup = AutofocusPopup(dummy_controller.view)
|
|
self.autofocus_controller = AutofocusPopupController(
|
|
autofocus_popup, dummy_controller
|
|
)
|
|
|
|
def test_init(self):
|
|
"""Tests that the controller is initialized correctly
|
|
|
|
Raises
|
|
------
|
|
AssertionError
|
|
If the controller is not initialized correctly
|
|
"""
|
|
assert isinstance(self.autofocus_controller, AutofocusPopupController)
|
|
assert self.autofocus_controller.view.popup.winfo_exists() == 1
|
|
|
|
def test_attr(self):
|
|
"""Tests that the attributes are initialized correctly
|
|
|
|
Raises
|
|
------
|
|
AssertionError
|
|
If the attributes are not initialized correctly
|
|
"""
|
|
|
|
# Listing off attributes to check existence
|
|
attrs = [
|
|
"autofocus_fig",
|
|
"autofocus_coarse",
|
|
"widgets",
|
|
"setting_dict",
|
|
]
|
|
|
|
for attr in attrs:
|
|
assert hasattr(self.autofocus_controller, attr)
|
|
|
|
def test_populate_experiment_values(self):
|
|
"""Tests that the values are populated correctly
|
|
|
|
Raises
|
|
------
|
|
AssertionError
|
|
If the values are not populated correctly
|
|
"""
|
|
microscope_name = self.autofocus_controller.microscope_name
|
|
device = self.autofocus_controller.widgets["device"].get()
|
|
device_ref = self.autofocus_controller.widgets["device_ref"].get()
|
|
for k in self.autofocus_controller.widgets:
|
|
if k != "device" and k != "device_ref":
|
|
assert self.autofocus_controller.widgets[k].get() == str(
|
|
self.autofocus_controller.setting_dict[microscope_name][device][
|
|
device_ref
|
|
][k]
|
|
)
|
|
# Some values are ints but Tkinter only uses strings
|
|
|
|
def test_update_experiment_values(self):
|
|
"""Tests that the values are updated correctly
|
|
|
|
Raises
|
|
------
|
|
AssertionError
|
|
If the values are not updated correctly
|
|
"""
|
|
# Changing values
|
|
self.autofocus_controller.widgets["coarse_range"].set(200)
|
|
self.autofocus_controller.widgets["coarse_step_size"].set(30)
|
|
self.autofocus_controller.view.setting_vars["coarse_selected"].set(False)
|
|
self.autofocus_controller.widgets["fine_range"].set(25)
|
|
self.autofocus_controller.widgets["fine_step_size"].set(2)
|
|
self.autofocus_controller.view.setting_vars["fine_selected"].set(False)
|
|
|
|
microscope_name = self.autofocus_controller.microscope_name
|
|
device = self.autofocus_controller.widgets["device"].get()
|
|
device_ref = self.autofocus_controller.widgets["device_ref"].get()
|
|
|
|
# Checking values match
|
|
for k in self.autofocus_controller.widgets:
|
|
if k != "device" and k != "device_ref":
|
|
assert self.autofocus_controller.widgets[k].get() == str(
|
|
self.autofocus_controller.setting_dict[microscope_name][device][
|
|
device_ref
|
|
][k]
|
|
)
|
|
for k in self.autofocus_controller.view.setting_vars:
|
|
assert (
|
|
self.autofocus_controller.view.setting_vars[k].get()
|
|
== self.autofocus_controller.setting_dict[microscope_name][device][
|
|
device_ref
|
|
][k]
|
|
)
|
|
|
|
def test_start_autofocus(self):
|
|
"""Tests that the start autofocus function works correctly
|
|
|
|
Raises
|
|
------
|
|
AssertionError
|
|
If the start autofocus function does not work correctly
|
|
"""
|
|
|
|
# Calling function
|
|
self.autofocus_controller.start_autofocus()
|
|
|
|
# Checking message sent
|
|
res = self.autofocus_controller.parent_controller.pop()
|
|
assert res == "autofocus"
|
|
|
|
self.autofocus_controller.parent_controller.clear()
|
|
|
|
def test_display_plot(self):
|
|
"""Tests that the display plot function works correctly
|
|
|
|
Todo: Retrieve data from axessubplot instance and
|
|
check that it is correct
|
|
|
|
Raises
|
|
------
|
|
AssertionError
|
|
If the display plot function does not work correctly
|
|
"""
|
|
# Make this robust by sending data and then
|
|
# checking each plot is plotting correct data low priority
|
|
|
|
x_data = np.linspace(start=69750.0, stop=70250.0, num=101)
|
|
y_data = np.random.rand(101)
|
|
data = [x_data, y_data]
|
|
self.autofocus_controller.display_plot([data, False, True])
|
|
pass
|