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)