def setUp(self):
super(SingletViewerTest, self).setUp()
self.canvas = Mock()
self.viewer = SingletViewer(self.rootframe, self.canvas)
self.do_render = Mock()
self.viewer._do_render = self.do_render
def setUp(self):
super(SingletViewerTest, self).setUp()
self.canvas = Mock()
self.viewer = SingletViewer(self.rootframe, self.canvas)
self.do_render = Mock()
self.viewer._do_render = self.do_render
def __init__(self, mainframe):
# Note: the figure we pass in is just a temporary placeholder.
# 'Displayable Items' provide their own figure which the canvas can
# be made to use, but it also requires one on its creation.
#self.canvas = FigureCanvas(mainframe.main_panel, wx.ID_ANY, plt.figure())
#mainframe.add_to_main_sizer(self.canvas)
self._singlet_viewer = SingletViewer(mainframe.main_panel)
self._triplet_viewer = TripletViewer(mainframe.main_panel)
self._image_viewer = self._singlet_viewer
def main():
app = wx.App()
rootframe = wx.Frame(None)
viewer = SingletViewer(rootframe)
hdulist = fits.open(context.get_test_data_path(TEST_FILE))
fits_image = Mock()
fits_image.as_hdulist.return_value = hdulist
viewer.display(fits_image)
viewer.draw_marker(50, 50, 10)
viewer.draw_marker(20, 70, 5)
rootframe.Show()
app.MainLoop()
def main():
app = wx.App()
rootframe = wx.Frame(None)
viewer = SingletViewer(rootframe)
hdulist = fits.open(context.get_test_data_path(TEST_FILE))
fits_image = Mock()
fits_image.as_hdulist.return_value = hdulist
viewer.display(fits_image)
viewer.draw_marker(50, 50, 10)
viewer.draw_marker(20, 70, 5)
rootframe.Show()
app.MainLoop()
class SingletViewerTest(WxWidgetTestCase):
def setUp(self):
super(SingletViewerTest, self).setUp()
self.canvas = Mock()
self.viewer = SingletViewer(self.rootframe, self.canvas)
self.do_render = Mock()
self.viewer._do_render = self.do_render
def test_mark_source(self):
x, y = 50, 100
fwhm = 4
cutout = Mock(spec=SourceCutout)
reading = Mock(spec=SourceReading)
reading.null_observation = Mock(return_value=False)
cutout.hdulist = mock_hdulist()
cutout.pixel_source_point = x, y
cutout.astrom_header = {"FWHM": fwhm}
reading.from_input_file = False
cutout.reading = reading
self.viewer.display(cutout, mark_source=False)
current_displayable = self.viewer.current_displayable
mock_place_marker = Mock()
current_displayable.place_marker = mock_place_marker
self.viewer.mark_sources(cutout)
mock_place_marker.assert_called_once_with(x, y, 2 * fwhm,
colour="b")
def test_refresh_marker(self):
cutout = Mock(spec=SourceCutout)
cutout.hdulist = mock_hdulist()
mark_source = Mock()
self.viewer.mark_sources = mark_source
self.viewer.display(cutout, mark_source=False)
self.viewer.refresh_markers()
mark_source.assert_called_once_with(cutout)
def test_toggle_reticule(self):
cutout = Mock(spec=SourceCutout)
cutout.hdulist = mock_hdulist()
mark_source = Mock()
self.viewer.mark_sources = mark_source
self.viewer.display(cutout)
toggle_reticule = Mock()
current_displayable = self.viewer.current_displayable
current_displayable.toggle_reticule = toggle_reticule
self.viewer.toggle_reticule()
toggle_reticule.assert_called_once_with()
class SingletViewerTest(WxWidgetTestCase):
def setUp(self):
super(SingletViewerTest, self).setUp()
self.canvas = Mock()
self.viewer = SingletViewer(self.rootframe, self.canvas)
self.do_render = Mock()
self.viewer._do_render = self.do_render
def test_mark_source(self):
x, y = 50, 100
fwhm = 4
cutout = Mock(spec=SourceCutout)
reading = Mock(spec=SourceReading)
reading.null_observation = Mock(return_value=False)
cutout.hdulist = mock_hdulist()
cutout.pixel_source_point = x, y
cutout.astrom_header = {"FWHM": fwhm}
reading.from_input_file = False
cutout.reading = reading
self.viewer.display(cutout, mark_source=False)
current_displayable = self.viewer.current_displayable
mock_place_marker = Mock()
current_displayable.place_marker = mock_place_marker
self.viewer.mark_sources(cutout)
mock_place_marker.assert_called_once_with(x, y, 2 * fwhm, colour="b")
def test_refresh_marker(self):
cutout = Mock(spec=SourceCutout)
cutout.hdulist = mock_hdulist()
mark_source = Mock()
self.viewer.mark_sources = mark_source
self.viewer.display(cutout, mark_source=False)
self.viewer.refresh_markers()
mark_source.assert_called_once_with(cutout)
def test_toggle_reticule(self):
cutout = Mock(spec=SourceCutout)
cutout.hdulist = mock_hdulist()
mark_source = Mock()
self.viewer.mark_sources = mark_source
self.viewer.display(cutout)
toggle_reticule = Mock()
current_displayable = self.viewer.current_displayable
current_displayable.toggle_reticule = toggle_reticule
self.viewer.toggle_reticule()
toggle_reticule.assert_called_once_with()
