class DisplayableImageSingletTest(unittest.TestCase): def setUp(self): self.singlet = DisplayableImageSinglet(MagicMock()) def mock_image_singlet(self): image_singlet = Mock(spec=ImageSinglet) self.singlet.image_singlet = image_singlet return image_singlet def test_reset_colormap(self): image_singlet = self.mock_image_singlet() self.singlet.reset_colormap() image_singlet.reset_colormap.assert_called_once_with() def test_toggle_reticule(self): image_singlet = self.mock_image_singlet() self.singlet.toggle_reticule() image_singlet.toggle_reticule.assert_called_once_with()
def setUp(self): self.singlet = DisplayableImageSinglet(MagicMock())
def display_hdulist(hdulist): DisplayableImageSinglet(hdulist).render()
def _create_displayable(self, cutout): return DisplayableImageSinglet(cutout.hdulist)