def test_to_spectrum1d():
# Set up simple spectral WCS
wcs = WCS(naxis=1)
wcs.wcs.ctype = ['VELO-LSR']
wcs.wcs.set()
coords = WCSCoordinates(wcs=wcs)
data = Data(label='spectrum', coords=coords)
data.add_component(Component(np.array([3.4, 2.3, -1.1, 0.3]), units='Jy'),
'x')
spec = data.get_object(Spectrum1D, attribute=data.id['x'])
assert_quantity_allclose(spec.spectral_axis, [1, 2, 3, 4] * u.m / u.s)
assert_quantity_allclose(spec.flux, [3.4, 2.3, -1.1, 0.3] * u.Jy)
data.add_subset(data.id['x'] > 1, label='bright')
spec_subset = data.get_subset_object(cls=Spectrum1D,
subset_id=0,
attribute=data.id['x'])
assert_quantity_allclose(spec_subset.spectral_axis,
[1, 2, 3, 4] * u.m / u.s)
assert_quantity_allclose(spec_subset.flux,
[3.4, 2.3, np.nan, np.nan] * u.Jy)
assert_equal(spec_subset.mask, [1, 1, 0, 0])
def test_to_ccddata(with_wcs):
if with_wcs:
coords = WCS_CELESTIAL
else:
coords = None
data = Data(label='image', coords=coords)
data.add_component(
Component(np.array([[3.4, 2.3], [-1.1, 0.3]]), units='Jy'), 'x')
image = data.get_object(CCDData, attribute=data.id['x'])
assert image.wcs is (WCS_CELESTIAL if with_wcs else None)
assert_allclose(image.data, [[3.4, 2.3], [-1.1, 0.3]])
assert image.unit is u.Jy
data.add_subset(data.id['x'] > 1, label='bright')
image_subset = data.get_subset_object(cls=CCDData,
subset_id=0,
attribute=data.id['x'])
assert image_subset.wcs is (WCS_CELESTIAL if with_wcs else None)
assert_allclose(image_subset.data, [[3.4, 2.3], [-1.1, 0.3]])
assert image_subset.unit is u.Jy
assert_equal(image_subset.mask, [[0, 0], [1, 1]])
def test_to_spectral_cube(spectral_cube_wcs):
data = Data(label='spectral_cube', coords=spectral_cube_wcs)
values = np.random.random((4, 5, 3))
data.add_component(Component(values, units='Jy'), 'x')
spec = data.get_object(SpectralCube, attribute=data.id['x'])
assert_quantity_allclose(spec.spectral_axis, [1, 2, 3, 4] * u.m / u.s)
assert_quantity_allclose(spec.filled_data[...], values * u.Jy)
data.add_subset(data.id['x'] > 0.5, label='bright')
spec_subset = data.get_subset_object(cls=SpectralCube, subset_id=0,
attribute=data.id['x'])
assert_quantity_allclose(spec_subset.spectral_axis, [1, 2, 3, 4] * u.m / u.s)
expected = values.copy()
expected[expected <= 0.5] = np.nan
assert_quantity_allclose(spec_subset.filled_data[...], expected * u.Jy)
assert_equal(spec_subset.mask.include(), values > 0.5)
class TestWWTDataViewer(object):
def setup_method(self, method):
self.d = Data(x=[1, 2, 3], y=[2, 3, 4], z=[4, 5, 6])
self.application = GlueApplication()
self.dc = self.application.data_collection
self.dc.append(self.d)
self.hub = self.dc.hub
self.session = self.application.session
self.viewer = self.application.new_data_viewer(WWTDataViewer)
self.options = self.viewer.options_widget()
def register(self):
self.viewer.register_to_hub(self.hub)
def test_add_data(self):
self.viewer.add_data(self.d)
self.viewer.state.lon_att = self.d.id['x']
self.viewer.state.lat_att = self.d.id['y']
def test_center(self):
self.viewer.add_data(self.d)
self.viewer.state.lon_att = self.d.id['x']
self.viewer.state.lat_att = self.d.id['y']
self.viewer.layers[0].center()
def test_new_subset_group(self):
# Make sure only the subset for data that is already inside the viewer
# is added.
d2 = Data(a=[4, 5, 6])
self.dc.append(d2)
self.viewer.add_data(self.d)
assert len(self.viewer.layers) == 1
self.dc.new_subset_group(subset_state=self.d.id['x'] > 1, label='A')
assert len(self.viewer.layers) == 2
def test_double_add_ignored(self):
assert len(self.viewer.layers) == 0
self.viewer.add_data(self.d)
assert len(self.viewer.layers) == 1
self.viewer.add_data(self.d)
assert len(self.viewer.layers) == 1
def test_remove_data(self):
self.register()
self.viewer.add_data(self.d)
layer = self.viewer._layer_artist_container[self.d][0]
layer.clear = MagicMock()
self.hub.broadcast(
message.DataCollectionDeleteMessage(self.dc, data=self.d))
# TODO: the following currently fails but is not critical, so we
# skip for now.
# assert layer.clear.call_count == 1
assert self.d not in self.viewer._layer_artist_container
def test_remove_subset(self):
self.register()
s = self.d.new_subset()
self.viewer.add_subset(s)
layer = self.viewer._layer_artist_container[s][0]
layer.clear = MagicMock()
self.hub.broadcast(message.SubsetDeleteMessage(s))
#assert layer.clear.call_count == 1
assert s not in self.viewer._layer_artist_container
def test_subsets_added_with_data(self):
s = self.d.new_subset()
self.viewer.add_data(self.d)
assert s in self.viewer._layer_artist_container
def test_subsets_live_added(self):
self.register()
self.viewer.add_data(self.d)
s = self.d.new_subset()
assert s in self.viewer._layer_artist_container
def test_clone(self):
self.viewer.add_data(self.d)
self.viewer.state.layers[0].ra_att = self.d.id['y']
self.viewer.state.layers[0].dec_att = self.d.id['x']
application2 = clone(self.application)
viewer2 = application2.viewers[0][0]
def test_changing_alt_back_to_none(self):
# Regression test for a bug which caused an exception to
# happen when changing the altitude attribute back to None
self.viewer.add_data(self.d)
self.viewer.state.mode = 'Milky Way'
self.viewer.state.lon_att = self.d.id['x']
self.viewer.state.lat_att = self.d.id['y']
self.viewer.state.alt_att = self.d.id['z']
self.viewer.state.alt_unit = 'kpc'
self.viewer.state.alt_att = None
def test_remove_layer(self):
# Make sure that _update_markers doesn't get called after removing a
# layer. This is a regression test for
# https://github.com/glue-viz/glue-wwt/issues/54
self.register()
self.d.add_subset(self.d.id['x'] > 1)
self.viewer.add_data(self.d)
assert len(self.viewer.layers) == 2
subset_layer = self.viewer.layers[1]
subset_layer.wwt_client.layers.add_table_layer = MagicMock()
self.viewer.remove_subset(self.d.subsets[0])
assert len(self.viewer.layers) == 1
assert subset_layer.wwt_client.layers.add_table_layer.call_count == 0
assert subset_layer.wwt_layer is None
class TestWWTDataViewer(object):
def setup_method(self, method):
self.d = Data(x=[1, 2, 3], y=[2, 3, 4], z=[4, 5, 6])
self.application = GlueApplication()
self.dc = self.application.data_collection
self.dc.append(self.d)
self.hub = self.dc.hub
self.session = self.application.session
self.viewer = self.application.new_data_viewer(WWTQtViewerBlocking)
self.options = self.viewer.options_widget()
def teardown_method(self, method):
self.viewer.close(warn=False)
self.viewer = None
self.application.close()
self.application = None
def register(self):
self.viewer.register_to_hub(self.hub)
def test_add_data(self):
self.viewer.add_data(self.d)
self.viewer.state.lon_att = self.d.id['x']
self.viewer.state.lat_att = self.d.id['y']
def test_center(self):
self.viewer.add_data(self.d)
self.viewer.state.lon_att = self.d.id['x']
self.viewer.state.lat_att = self.d.id['y']
self.viewer.layers[0].center()
def test_new_subset_group(self):
# Make sure only the subset for data that is already inside the viewer
# is added.
d2 = Data(a=[4, 5, 6])
self.dc.append(d2)
self.viewer.add_data(self.d)
assert len(self.viewer.layers) == 1
self.dc.new_subset_group(subset_state=self.d.id['x'] > 1, label='A')
assert len(self.viewer.layers) == 2
def test_double_add_ignored(self):
assert len(self.viewer.layers) == 0
self.viewer.add_data(self.d)
assert len(self.viewer.layers) == 1
self.viewer.add_data(self.d)
assert len(self.viewer.layers) == 1
def test_remove_data(self):
self.register()
self.viewer.add_data(self.d)
layer = self.viewer._layer_artist_container[self.d][0]
layer.clear = MagicMock()
self.hub.broadcast(
message.DataCollectionDeleteMessage(self.dc, data=self.d))
# TODO: the following currently fails but is not critical, so we
# skip for now.
# assert layer.clear.call_count == 1
assert self.d not in self.viewer._layer_artist_container
def test_remove_subset(self):
self.register()
s = self.d.new_subset()
self.viewer.add_subset(s)
layer = self.viewer._layer_artist_container[s][0]
layer.clear = MagicMock()
self.hub.broadcast(message.SubsetDeleteMessage(s))
# assert layer.clear.call_count == 1
assert s not in self.viewer._layer_artist_container
def test_subsets_added_with_data(self):
s = self.d.new_subset()
self.viewer.add_data(self.d)
assert s in self.viewer._layer_artist_container
def test_subsets_live_added(self):
self.register()
self.viewer.add_data(self.d)
s = self.d.new_subset()
assert s in self.viewer._layer_artist_container
def test_clone(self):
self.viewer.add_data(self.d)
self.viewer.state.layers[0].ra_att = self.d.id['y']
self.viewer.state.layers[0].dec_att = self.d.id['x']
application2 = clone(self.application)
application2.viewers[0][0]
def test_changing_alt_back_to_none(self):
# Regression test for a bug which caused an exception to
# happen when changing the altitude attribute back to None
self.viewer.add_data(self.d)
self.viewer.state.mode = 'Milky Way'
self.viewer.state.lon_att = self.d.id['x']
self.viewer.state.lat_att = self.d.id['y']
self.viewer.state.alt_att = self.d.id['z']
self.viewer.state.alt_unit = 'kpc'
self.viewer.state.alt_att = None
def test_remove_layer(self):
# Make sure that _update_markers doesn't get called after removing a
# layer. This is a regression test for
# https://github.com/glue-viz/glue-wwt/issues/54
self.register()
self.d.add_subset(self.d.id['x'] > 1)
self.viewer.add_data(self.d)
assert len(self.viewer.layers) == 2
subset_layer = self.viewer.layers[1]
subset_layer.wwt_client.layers.add_table_layer = MagicMock()
self.viewer.remove_subset(self.d.subsets[0])
assert len(self.viewer.layers) == 1
assert subset_layer.wwt_client.layers.add_table_layer.call_count == 0
assert subset_layer.wwt_layer is None
@pytest.mark.skipif(sys.platform == 'win32',
reason="Test causes issues on Windows")
def test_save_tour(self, tmpdir):
filename = tmpdir.join('mytour.wtt').strpath
self.viewer.add_data(self.d)
with patch.object(compat,
'getsavefilename',
return_value=(filename, None)):
self.viewer.toolbar.tools['save'].subtools[1].activate()
assert os.path.exists(filename)
with io.open(filename, newline='') as f:
assert f.read().startswith(
"<?xml version='1.0' encoding='UTF-8'?>\r\n<FileCabinet")
def test_load_session_back_compat(self):
# Make sure that old session files continue to work
app = GlueApplication.restore_session(
os.path.join(DATA, 'wwt_simple.glu'))
viewer_state = app.viewers[0][0].state
assert viewer_state.lon_att.label == 'a'
assert viewer_state.lat_att.label == 'b'
assert viewer_state.frame == 'Galactic'