fixup

Author: JackEAllen

mantidimaging/eyes_tests/live_viewer_window_test.py

@@ -1,19 +1,15 @@
 # Copyright (C) 2024 ISIS Rutherford Appleton Laboratory UKRI
 # SPDX - License - Identifier: GPL-3.0-or-later
-from __future__ import annotations
 
+from __future__ import annotations
 from typing import TYPE_CHECKING
 from unittest import mock
-
 import numpy as np
-
 import os
-
 from mantidimaging.core.operations.loader import load_filter_packages
 from mantidimaging.gui.windows.live_viewer.model import Image_Data
 from mantidimaging.test_helpers.unit_test_helper import FakeFSTestCase
 from pathlib import Path
-
 from mantidimaging.eyes_tests.base_eyes import BaseEyesTest
 
 if TYPE_CHECKING:
@@ -22,6 +18,10 @@
 
 class LiveViewerWindowTest(FakeFSTestCase, BaseEyesTest):
 
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.initial_time = 4000.0
+
     @classmethod
     def setUpClass(cls) -> None:
         super().setUpClass()
@@ -32,7 +32,6 @@ def setUp(self) -> None:
         self.fs.add_real_directory(Path(__file__).parent.parent)  # Allow ui file to be found
         self.live_directory = Path("/live_dir")
         self.fs.create_dir(self.live_directory)
-        self.initial_time = 4000.0
 
     def _generate_image(self):
         image = np.zeros((10, 10))
@@ -43,16 +42,11 @@ def _generate_image(self):
 
     def _make_simple_dir(self, directory: Path):
         file_list = [directory / f"abc_{i:06d}.tif" for i in range(5)]
-        # set initial time to live directory utime
-        # initial_time = 4000.0
-        if not directory.exists():
-            self.fs.create_dir(directory)
-        # increment = 1
+        increment = 0
         for file in file_list:
             self.fs.create_file(file)
-            # set file time to incrememnt by 1
-            # os.utime(file, (10, initial_time + increment))
-            # increment += 1
+            os.utime(file, (10, self.initial_time + increment))
+            increment += 1000
 
         return file_list
 
@@ -100,15 +94,11 @@ def test_rotate_operation_rotates_image(self, _mock_time, _mock_image_watcher, m
     @mock.patch('mantidimaging.gui.windows.live_viewer.presenter.LiveViewerWindowPresenter.load_image')
     @mock.patch('mantidimaging.gui.windows.live_viewer.model.ImageWatcher')
     @mock.patch("time.time", return_value=4000.0)
-    def test_file_time_is_updated(self, _mock_time, _mock_image_watcher, mock_load_image):
+    def test_file_time_lebel_displays_correct_time(self, _mock_time, _mock_image_watcher, mock_load_image):
         file_list = self._make_simple_dir(self.live_directory)
         image_list = [Image_Data(path) for path in file_list]
-        increment = 0
-        for image in image_list:
-            image.image_modified_time_stamp = 4000.0 + increment
-            mock_load_image.return_value = self._generate_image()
-            self.imaging.show_live_viewer(self.live_directory)
-            self.assertEqual(image.image_modified_time_stamp,
-                             4000.0 + increment)  # Add this line to check the label time
-            self.imaging.live_viewer.presenter.model._handle_image_changed_in_list(image_list)
-            self.check_target(widget=self.imaging.live_viewer)
+        mock_load_image.return_value = self._generate_image()
+        self.imaging.show_live_viewer(self.live_directory)
+        self.imaging.live_viewer.presenter.model._handle_image_changed_in_list(image_list)
+        self.assertEqual(self.imaging.live_viewer.file_time_label.text(), "1970-01-01 02:13:20")
+        self.check_target(widget=self.imaging.live_viewer)