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)