test GUI file for GUI ca exp creator

news/test_gui_ca_exp_creator.rst

@@ -0,0 +1,23 @@
+**Added:**
+
+* Added pytest code for gui_ca_exp_creator
+
+**Changed:**
+
+* <news item>
+
+**Deprecated:**
+
+* <news item>
+
+**Removed:**
+
+* <news item>
+
+**Fixed:**
+
+* <news item>
+
+**Security:**
+
+* <news item>

tests/test_gui_ca_exp_creator.py

@@ -0,0 +1,271 @@
+import os
+import tempfile
+from unittest.mock import MagicMock
+
+# import numpy as np
+import pytest
+
+# CA_main,; plot,; preview_CA,; write_to_file_CA,
+from src.GUI.code.GUI_CA_exp_creator import (
+    CA_window_writer,
+    check,
+    config_writer_CA,
+    get_AP_parameters,
+    get_parameters,
+    load_APwindow,
+    load_folder,
+    save_file,
+    time_converter,
+)
+
+# Sample data for testing
+sample_AP_params = [4.798, 1.03, 0, 0.202, 0.003, 9]
+sample_exp_params = {
+    "experiment_file_name": "test_output",
+    "experiment_file_path": tempfile.gettempdir(),
+    "experiment_rest_time": "10",
+    "experiment_step_number": "5",
+    "experiment_voltage": "0.5",
+    "experiment_time": "60",
+}
+
+
+# Mocking the UI class
+class MockUiCA:
+    def __init__(self):
+        self.experiment_file_path = MagicMock()
+        self.load_folder_name = MagicMock(return_value=tempfile.gettempdir())
+        self.AP_window = MagicMock(return_value=(MagicMock(), MagicMock()))
+        self.experiment_queue = MagicMock()
+        self.plot_area = MagicMock()
+        self.save_experiment_file_button = MagicMock()
+
+
+class MockUiExperiment:
+    def __init__(self):
+        self.experiment_file_name = MagicMock()
+        self.experiment_file_path = MagicMock()
+        self.experiment_rest_time = MagicMock()
+        self.experiment_step_number = MagicMock()
+        self.experiment_voltage = MagicMock()
+        self.experiment_time = MagicMock()
+        self.experiment_duration = MagicMock()
+        self.experiment_averag_number = MagicMock()
+        self.experiment_conversion_factor = MagicMock()
+        self.experiment_setpoint_gain = MagicMock()
+        self.experiment_setpoint_offset = MagicMock()
+        self.experiment_shunt_resistor = MagicMock()
+        self.experiment_time_step = MagicMock()
+
+
+# Test for load_folder function
+def test_load_folder():
+    CA = MockUiCA()
+    load_folder(CA)
+    assert CA.experiment_file_path.setText.called
+
+
+# Test for get_AP_parameters function
+def test_get_AP_parameters():
+    exp = MockUiExperiment()
+    exp.experiment_conversion_factor.text.return_value = "1.0"
+    exp.experiment_setpoint_gain.text.return_value = "2.0"
+    exp.experiment_setpoint_offset.text.return_value = "0.1"
+    exp.experiment_shunt_resistor.text.return_value = "0.202"
+    exp.experiment_time_step.text.return_value = "0.003"
+    exp.experiment_averag_number.text.return_value = "9"
+
+    AP_parameters = get_AP_parameters(exp)
+    assert AP_parameters == [1.0, 2.0, 0.1, 0.202, 0.003, 9]
+
+
+# Test for load_APwindow function
+def test_load_APwindow():
+    CA = MockUiCA()
+    AP_params = sample_AP_params
+    AP_window_mock, _ = CA.AP_window()
+
+    load_APwindow(CA, AP_params)
+
+    assert AP_window_mock.experiment_conversion_factor.setText.called
+    assert AP_window_mock.experiment_setpoint_gain.setText.called
+    assert AP_window_mock.experiment_setpoint_offset.setText.called
+    assert AP_window_mock.experiment_shunt_resistor.setText.called
+    assert AP_window_mock.experiment_time_step.setText.called
+    assert AP_window_mock.experiment_averag_number.setText.called
+
+    # Check if the correct values were set
+    assert AP_window_mock.experiment_conversion_factor.setText.call_args[0][
+        0
+    ] == str(AP_params[0])
+    assert AP_window_mock.experiment_setpoint_gain.setText.call_args[0][
+        0
+    ] == str(AP_params[1])
+    assert AP_window_mock.experiment_setpoint_offset.setText.call_args[0][
+        0
+    ] == str(AP_params[2])
+    assert AP_window_mock.experiment_shunt_resistor.setText.call_args[0][
+        0
+    ] == str(AP_params[3])
+    assert AP_window_mock.experiment_time_step.setText.call_args[0][0] == str(
+        AP_params[4]
+    )
+    assert AP_window_mock.experiment_averag_number.setText.call_args[0][
+        0
+    ] == str(AP_params[5])
+
+
+# Test for time_converter function
+def test_time_converter():
+    assert time_converter(3661) == "1:01:01"
+    assert time_converter(-1) == "0:00:00"
+
+
+# Test for preview_CA function
+# def test_preview_CA():
+#     # Create a mock Ui_experiment object
+#     exp = MagicMock()
+#     exp.experiment_step_number.text.return_value = '10'
+#     exp.experiment_voltage.text.return_value = '5.0'
+#     exp.experiment_time.text.return_value = '60'
+#     exp.experiment_duration.setText = MagicMock()
+#
+#     # Create a mock QGridLayout object
+#     grid = MagicMock()
+#
+#     # Call the function
+#     preview_CA(exp, grid)
+#
+#     # Check if the plot function was called with the correct parameters
+#     expected_times=np.linspace(0, 60, num=11)
+#     expected_steps=np.linspace(5.0, 5.0, 11)
+#     grid.addWidget.assert_called_once()
+#     assert exp.experiment_duration.setText.call_count == 1
+
+
+# Test for save_file function
+def test_save_file():
+    config_data = {"test_key": "test_value"}
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        config_path_name = temp_file.name
+    save_file(config_path_name, config_data, 0)
+
+    with open(config_path_name, "r") as file:
+        contents = file.read()
+    assert "#General parameters to edit for an experiment" in contents
+    os.remove(config_path_name)
+
+
+# Test for config_writer_CA function
+def test_config_writer_CA():
+    exp_type = "CA"
+    rt = 10
+    stp_num = 5
+    data_out_name = "test_output"
+    data_out_path = tempfile.gettempdir()
+    conv_fact = 1.0
+    set_gain = 2.0
+    set_offset = 0.1
+    sr = 0.202
+    ts = 0.003
+    avg_num = 9
+    ca_v = 0.5
+    ca_time = 60
+
+    config_writer_CA(
+        exp_type,
+        rt,
+        stp_num,
+        data_out_name,
+        data_out_path,
+        conv_fact,
+        set_gain,
+        set_offset,
+        sr,
+        ts,
+        avg_num,
+        ca_v,
+        ca_time,
+    )
+    config_file_path = os.path.join(
+        data_out_path, f"{data_out_name}_{exp_type}_config.yml"
+    )
+    assert os.path.exists(config_file_path)
+    os.remove(config_file_path)
+
+
+# Test for check function
+def test_check():
+    assert check("123") is True
+    assert check("-123") is True
+    assert check("12.3") is True
+    assert check("abc") is False
+    assert check("") is False
+
+
+# Test for get_parameters function
+def test_get_parameters():
+    exp = MockUiExperiment()
+    exp.experiment_file_name.text.return_value = "test_output"
+    exp.experiment_file_path.text.return_value = tempfile.gettempdir()
+    exp.experiment_rest_time.text.return_value = "10"
+    exp.experiment_step_number.text.return_value = "5"
+    exp.experiment_voltage.text.return_value = "0.5"
+    exp.experiment_time.text.return_value = "60"
+
+    AP_parameters = sample_AP_params
+    config_params = get_parameters(exp, "CA", AP_parameters)
+    assert config_params[0] == "CA"
+    assert config_params[3] == "test_output"
+
+
+# Test for write_to_file_CA function
+# def test_write_to_file_CA():
+#     ui = MagicMock()
+#     CA = MockUiCA()
+#     CA_window = MagicMock()
+#     AP_params = sample_AP_params
+#
+#     write_to_file_CA(ui, CA, CA_window, AP_params)
+#     assert CA_window.close.called
+
+
+# Test for CA_window_writer function
+def test_CA_window_writer():
+    exp = MockUiExperiment()
+    config_data = {
+        "general_parameters": {
+            "data_output_filename": "test_output",
+            "data_output_path": tempfile.gettempdir(),
+            "rest_time": 10,
+            "step_number": 5,
+        },
+        "chronoamperometry": {
+            "voltage": 0.5,
+            "time": 60,
+        },
+    }
+
+    CA_window_writer(exp, config_data)
+    assert exp.experiment_file_name.setText.called
+    assert exp.experiment_file_path.setText.called
+
+
+# Test for CA_main function
+# def test_CA_main():
+#     ui = MagicMock()
+#     config_data = None
+#     CA = CA_main(ui, config_data)
+#     assert CA is not None
+
+
+# Test for plot function
+# def test_plot():
+#     x = [0, 1, 2]
+#     y = [0, 1, 0]
+#     canvas = plot(x, y)
+#     assert canvas is not None
+
+
+if __name__ == "__main__":
+    pytest.main()