def __init__(self, line_interaction_analysis, atom_data, description): super(LineInteractionTables, self).__init__() self.text_description = QtGui.QLabel(str(description)) self.species_table = QtGui.QTableView() self.transitions_table = QtGui.QTableView() self.layout = QtGui.QHBoxLayout() self.line_interaction_analysis = line_interaction_analysis self.atom_data = atom_data line_interaction_species_group = line_interaction_analysis.last_line_in.groupby(['atomic_number', 'ion_number']) self.species_selected = sorted(line_interaction_species_group.groups.keys()) species_symbols = [util.species_tuple_to_string(item, atom_data) for item in self.species_selected] species_table_model = SimpleTableModel([species_symbols, ['Species']]) species_abundances = (line_interaction_species_group.wavelength.count().astype(float) / line_interaction_analysis.last_line_in.wavelength.count()).astype(float).tolist() species_abundances = map(float, species_abundances) species_table_model.addData(species_abundances) self.species_table.setModel(species_table_model) line_interaction_species_group.wavelength.count() self.layout.addWidget(self.text_description) self.layout.addWidget(self.species_table) self.species_table.connect(self.species_table.verticalHeader(), QtCore.SIGNAL('sectionClicked(int)'), self.on_species_clicked) self.layout.addWidget(self.transitions_table) self.setLayout(self.layout) self.show()
def test_species_tuple_to_species_string(species_string, species_tuple): assert species_tuple_to_string(species_tuple) == species_string
def test_species_tuple_to_string(species_tuple, roman_numerals, species_string): assert species_tuple_to_string( species_tuple, roman_numerals=roman_numerals) == species_string
def test_species_tuple_to_string(species_tuple, roman_numerals, species_string): assert species_tuple_to_string(species_tuple, roman_numerals=roman_numerals) == species_string