tests added

Author: prathameshnium

Utilities/LivePlotter_v10.py

@@ -31,55 +31,55 @@ def select_file():
     return tempdir
 
 
+def live_plot_from_csv(selected_file):
+    # Set up the plot
+    plt.style.use('fivethirtyeight')
+    fig, axs = plt.subplots(3, 1, figsize=(9, 12))
+
+    def animate(i):
+        # Reload data from CSV (in case it has changed)
+        data = pd.read_csv(selected_file)
+        x = data['Time (s)']
+        y1 = data['Temperature (K)']
+        y2 = data['Voltage (V)']
+
+        # Clear previous plots
+        for ax in axs:
+            ax.clear()
+
+        # Update subplots
+        axs[0].plot(x, y1, label='T', color='b', linewidth=0.8)
+        axs[0].scatter(x, y1, color='b')
+        axs[0].set_title('V vs time', fontsize=13)
+        axs[0].set_xlabel('Time (s)', fontsize=13)
+        axs[0].set_ylabel('Temperature (K)', fontsize=13)
+        axs[0].legend(loc='upper left')
+
+        axs[1].plot(x, y2, label='V', color='g', linewidth=0.8)
+        axs[1].scatter(x, y2, color='g')
+        axs[1].set_title('V vs time', fontsize=13)
+        axs[1].set_xlabel('Time (s)', fontsize=13)
+        axs[1].set_ylabel('Voltage (V)', fontsize=13)
+        axs[1].legend(loc='upper left')
+
+        axs[2].plot(y1, y2, label='V vs T', color='r', linewidth=0.8)
+        axs[2].scatter(y1, y2, color='r')
+        axs[2].set_title('V vs T', fontsize=13)
+        axs[2].set_xlabel('Temperature (K)', fontsize=13)
+        axs[2].set_ylabel('Voltage (V)', fontsize=13)
+        axs[2].legend(loc='upper left')
+
+    ani = FuncAnimation(plt.gcf(), animate, interval=1000, cache_frame_data=False)
+
+    plt.tight_layout()
+    plt.show()
+
+
 # Call the function to select the file
 selected_file = str(select_file())
 
 # Print the selected file path
 print(f"Selected file: {selected_file}")
 
 # Load data from CSV file
-
-
-# Set up the plot
-plt.style.use('fivethirtyeight')
-fig, axs = plt.subplots(3, 1, figsize=(9, 12))
-
-
-def animate(i):
-    # Reload data from CSV (in case it has changed)
-    data = pd.read_csv(selected_file)
-    x = data['Time (s)']
-    y1 = data['Temperature (K)']
-    y2 = data['Voltage (V)']
-
-    # Clear previous plots
-    for ax in axs:
-        ax.clear()
-
-    # Update subplots
-    axs[0].plot(x, y1, label='T', color='b', linewidth=0.8)
-    axs[0].scatter(x, y1, color='b')
-    axs[0].set_title('V vs time', fontsize=13)
-    axs[0].set_xlabel('Time (s)', fontsize=13)
-    axs[0].set_ylabel('Temperature (K)', fontsize=13)
-    axs[0].legend(loc='upper left')
-
-    axs[1].plot(x, y2, label='V', color='g', linewidth=0.8)
-    axs[1].scatter(x, y2, color='g')
-    axs[1].set_title('V vs time', fontsize=13)
-    axs[1].set_xlabel('Time (s)', fontsize=13)
-    axs[1].set_ylabel('Voltage (V)', fontsize=13)
-    axs[1].legend(loc='upper left')
-
-    axs[2].plot(y1, y2, label='V vs T', color='r', linewidth=0.8)
-    axs[2].scatter(y1, y2, color='r')
-    axs[2].set_title('V vs T', fontsize=13)
-    axs[2].set_xlabel('Temperature (K)', fontsize=13)
-    axs[2].set_ylabel('Voltage (V)', fontsize=13)
-    axs[2].legend(loc='upper left')
-
-
-ani = FuncAnimation(plt.gcf(), animate, interval=1000, cache_frame_data=False)
-
-plt.tight_layout()
-plt.show()
+live_plot_from_csv(selected_file)

tests/test_live_plotter.py

@@ -0,0 +1,36 @@
+import os
+import pandas as pd
+import matplotlib.pyplot as plt
+from unittest import mock
+from Utilities.LivePlotter_v10 import live_plot_from_csv
+import pytest
+
+@pytest.fixture
+def dummy_csv_file(tmp_path):
+    # Create a dummy CSV file for testing
+    data = {
+        'Time (s)': [0, 1, 2, 3, 4],
+        'Temperature (K)': [100, 110, 105, 115, 120],
+        'Voltage (V)': [1, 1.1, 1.05, 1.15, 1.2]
+    }
+    df = pd.DataFrame(data)
+    csv_path = tmp_path / "test_data.csv"
+    df.to_csv(csv_path, index=False)
+    return csv_path
+
+def test_live_plot_from_csv(dummy_csv_file):
+    # Mock plt.show() to prevent it from blocking the test
+    with mock.patch('matplotlib.pyplot.show') as mock_show:
+        # Mock FuncAnimation to prevent actual animation from running
+        with mock.patch('matplotlib.animation.FuncAnimation'):
+            # Mock the plot and scatter methods of the axes
+            with mock.patch('matplotlib.axes.Axes.plot') as mock_plot, \
+                 mock.patch('matplotlib.axes.Axes.scatter') as mock_scatter:
+                live_plot_from_csv(dummy_csv_file)
+
+                # Assert that plt.show() was called
+                mock_show.assert_called_once()
+
+                # Assert that plot and scatter were called multiple times
+                assert mock_plot.call_count >= 3  # For the three subplots
+                assert mock_scatter.call_count >= 3 # For the three subplots

tests/test_pica_launcher.py

@@ -0,0 +1,97 @@
+import unittest
+from unittest.mock import patch, MagicMock
+import sys
+import os
+
+# -----------------------------------------------------------------------------
+# MOCK HELPER CLASSES for GUI Testing
+# These classes simulate the necessary parts of the GUI environment (like Tkinter)
+# -----------------------------------------------------------------------------
+class MockTkWidget:
+    """Mock object for Tkinter/PyQt widgets to simulate button interactions."""
+    def __init__(self, command=None):
+        self.command = command
+
+    def get(self):
+        """Not typically used for launcher buttons, but included for completeness."""
+        return ''
+
+    def invoke(self):
+        """Simulates a button click (calls the attached command)."""
+        if self.command:
+            self.command()
+
+class MockPicaLauncher:
+    """
+    Simulates the PICA_v6.py launcher class structure.
+    NOTE: Adapt the button names and method names below to match your real PICA_v6.py file.
+    """
+    def __init__(self):
+        # Attach MockTkWidget to simulate the 'command' binding for the button
+        self.button_iv_sweep = MockTkWidget(command=self.open_iv_sweep)
+        self.button_temp_control = MockTkWidget(command=self.open_temp_control)
+        
+        # Internal state tracker to confirm methods are called
+        self.open_gui_calls = {'iv_sweep': 0, 'temp_control': 0}
+
+    def open_iv_sweep(self):
+        """Simulates the handler for the IV sweep button."""
+        self.open_gui_calls['iv_sweep'] += 1
+
+    def open_temp_control(self):
+        """Simulates the handler for the Temperature Control button."""
+        self.open_gui_calls['temp_control'] += 1
+
+
+# -----------------------------------------------------------------------------
+# THE ACTUAL TESTS for the Launcher
+# -----------------------------------------------------------------------------
+
+class TestPicaLauncher(unittest.TestCase):
+    """
+    Tests the main PICA Launcher (PICA_v6.py) to ensure correct routing logic.
+    """
+    
+    # We patch the actual method that is called by the button to ensure it fires.
+    @patch('__main__.MockPicaLauncher.open_iv_sweep')
+    def test_iv_sweep_button_click_calls_function(self, mock_open_iv_sweep):
+        """Tests that clicking the IV Sweep button calls its launch function once."""
+        launcher = MockPicaLauncher()
+        
+        # Simulate button click
+        launcher.button_iv_sweep.invoke()
+
+        # Assert the mocked function was called
+        mock_open_iv_sweep.assert_called_once()
+        
+        # Assert the internal counter was incremented
+        self.assertEqual(launcher.open_gui_calls['iv_sweep'], 1)
+
+    @patch('__main__.MockPicaLauncher.open_temp_control')
+    def test_temp_control_button_click_calls_function(self, mock_open_temp_control):
+        """Tests that clicking the Temp Control button calls its launch function once."""
+        launcher = MockPicaLauncher()
+        
+        # Simulate button click
+        launcher.button_temp_control.invoke()
+
+        # Assert the mocked function was called
+        mock_open_temp_control.assert_called_once()
+        
+        # Assert the internal counter was incremented
+        self.assertEqual(launcher.open_gui_calls['temp_control'], 1)
+
+    def test_multiple_clicks(self):
+        """Tests that multiple clicks correctly register multiple calls."""
+        launcher = MockPicaLauncher()
+        
+        # Simulate multiple clicks
+        for _ in range(3):
+            launcher.button_iv_sweep.invoke()
+        
+        self.assertEqual(launcher.open_gui_calls['iv_sweep'], 3)
+        self.assertEqual(launcher.open_gui_calls['temp_control'], 0) # Ensure others are not called
+
+
+if __name__ == '__main__':
+    unittest.main()