def test_exporter_cancel(tmpdir): filename = tmpdir.join('test.fits').strpath data = Data(x=[1, 2, 3]) data_collection = DataCollection([data]) data_collection.new_subset_group(subset_state=data.id['x'] >= 2, label='subset a') with patch('qtpy.compat.getsavefilename') as o: o.return_value = '', '' # simulates cancelling exporter = QtSubsetMaskExporter() exporter.run(data) assert not os.path.exists(filename)
def test_exporter(tmpdir): filename = tmpdir.join('test.fits').strpath data = Data(x=[1, 2, 3]) data_collection = DataCollection([data]) data_collection.new_subset_group(subset_state=data.id['x'] >= 2, label='subset a') with patch('qtpy.compat.getsavefilename') as o: o.return_value = filename, 'FITS (*.fits *.fit *.fits.gz *.fit.gz)' exporter = QtSubsetMaskExporter() exporter.run(data) with fits.open(filename) as hdulist: assert len(hdulist) == 2 assert hdulist[0].data is None assert hdulist[1].name == 'SUBSET A' assert_equal(hdulist[1].data, [0, 1, 1])
def _do_action(self): assert self._can_trigger() data = self.selected_layers()[0] from glue.io.qt.subset_mask import QtSubsetMaskExporter QtSubsetMaskExporter().run(data)