diff --git a/tests/test_work_unit.py b/tests/test_work_unit.py index 5a93d592..b5ea19a8 100644 --- a/tests/test_work_unit.py +++ b/tests/test_work_unit.py @@ -66,6 +66,7 @@ def test_save_and_load_fits(self): li = work2.im_stack.get_single_image(i) self.assertEqual(li.get_width(), self.width) self.assertEqual(li.get_height(), self.height) + self.assertEqual(li.get_obstime(), 2 * i + 1) # Check the three image layers match. sci1 = li.get_science()