def test_importer(tmpdir):
filename = tmpdir.join('test.fits').strpath
hdu = fits.PrimaryHDU(data=[0, 1, 1])
hdu.writeto(filename)
data = Data(x=[1, 2, 3])
data_collection = DataCollection([data])
with patch('qtpy.compat.getopenfilename') as o:
o.return_value = filename, 'FITS (*.fits *.fit *.fits.gz *.fit.gz)'
importer = QtSubsetMaskImporter()
importer.run(data, data_collection)
assert_equal(data.subsets[0].to_mask(), [0, 1, 1])
def test_importer_cancel(tmpdir):
filename = tmpdir.join('test.fits').strpath
hdu = fits.PrimaryHDU(data=[0, 1, 1])
hdu.writeto(filename)
data = Data(x=[1, 2, 3])
data_collection = DataCollection([data])
with patch('qtpy.compat.getopenfilename') as o:
o.return_value = '', '' # simulates cancelling
importer = QtSubsetMaskImporter()
importer.run(data, data_collection)
assert len(data_collection.subset_groups) == 0
assert len(data.subsets) == 0
def _do_action(self):
assert self._can_trigger()
data = self.selected_layers()[0]
from glue.io.qt.subset_mask import QtSubsetMaskImporter
QtSubsetMaskImporter().run(data, self._layer_tree._data_collection)