def test_scatter_remove_layer_artists(tmpdir): # Regression test for a bug that caused layer states to not be removed # when the matching layer artist was removed. This then caused issues when # loading session files. # Create fake data data = make_test_data() # Create fake session dc = DataCollection([data]) ga = GlueApplication(dc) ga.show() scatter = ga.new_data_viewer(VispyScatterViewer) scatter.add_data(data) dc.new_subset_group(subset_state=data.id['x'] > 0.5, label='subset 1') scatter.add_subset(data.subsets[0]) assert len(scatter.layers) == 2 assert len(scatter.state.layers) == 2 dc.remove_subset_group(dc.subset_groups[0]) assert len(scatter.layers) == 1 assert len(scatter.state.layers) == 1 # Check that writing a session works as expected. session_file = tmpdir.join('test_scatter_viewer.glu').strpath ga.save_session(session_file) ga.close() # Now we can check that everything is restored correctly ga2 = GlueApplication.restore_session(session_file) ga2.show() ga2.close()
def test_scatter_remove_layer_artists(tmpdir): # Regression test for a bug that caused layer states to not be removed # when the matching layer artist was removed. This then caused issues when # loading session files. # Create fake data data = make_test_data() # Create fake session dc = DataCollection([data]) ga = GlueApplication(dc) ga.show() scatter = ga.new_data_viewer(VispyScatterViewer) scatter.add_data(data) dc.new_subset_group(subset_state=data.id['x'] > 0.5, label='subset 1') scatter.add_subset(data.subsets[0]) assert len(scatter.layers) == 2 assert len(scatter.state.layers) == 2 dc.remove_subset_group(dc.subset_groups[0]) assert len(scatter.layers) == 1 assert len(scatter.state.layers) == 1 # Check that writing a session works as expected. session_file = tmpdir.join('test_scatter_viewer.glu').strpath ga.save_session(session_file) ga.close() # Now we can check that everything is restored correctly ga2 = GlueApplication.restore_session(session_file) ga2.show()
def test_remove_subset_group(): # Regression test for a bug that meant that removing a subset caused an # error when multiple viewers were present. # Create fake data data = make_test_data() # Create fake session dc = DataCollection([data]) ga = GlueApplication(dc) ga.show() volume1 = ga.new_data_viewer(VispyVolumeViewer) volume1.add_data(data) volume2 = ga.new_data_viewer(VispyVolumeViewer) volume2.add_data(data) dc.new_subset_group(subset_state=data.id['a'] > 0, label='Subset 1') dc.remove_subset_group(dc.subset_groups[0]) ga.close()