def show_image(self, colorbar=False): # start xpans if needed ds9.ds9_xpans() # start ds9 if need, or connect to existing display = ds9.ds9(target='validate') if self.frame_number is None: # display.set('frame delete all') display.set('frame new') display.set('scale zscale') display.set('cmap invert yes') f = StringIO() self.hdulist.writeto(f) f.flush() f.seek(0) hdulist = fits.open(f) for hdu in hdulist: del(hdu.header['PV*']) display.set_pyfits(hdulist) f.close() del(hdulist) self.frame_number = display.get('frame frameno') display.set('frame center {}'.format(self.frame_number)) display.set('zoom to fit') display.set('wcs align yes') display.set('frame frameno {}'.format(self.frame_number)) self._interaction_context = InteractionContext(self) self.number_of_images_displayed += 1
def setUp(self): mainhdu = Mock() mainhdu.data.shape = (100, 100) self.hdulist = [mainhdu] self.figure = Mock() with patch.object(ImageSinglet, "_create_axes"): self.image = ImageSinglet(self.hdulist, self.figure, [0, 0, 1, 1]) self.image.axes = Mock() self.interaction_context = InteractionContext(self.image)