def test_atom_data_two_instances_same_session(test_session): atom_data1 = AtomData(test_session, selected_atoms="He, Be, B, N, Zn", chianti_ions="He 1; N 5") atom_data2 = AtomData(test_session, selected_atoms="He, Be, B, N, Zn", chianti_ions="He 1; N 5") atom_data1.chianti_ions_table atom_data2.chianti_ions_table
def test_atom_data_chianti_ions_subset(memory_session): nist = DataSource.as_unique(memory_session, short_name="nist-asd") ch = DataSource.as_unique(memory_session, short_name="chianti_v8.0.2") ku = DataSource.as_unique(memory_session, short_name="ku_latest") with pytest.raises(ValueError): atom_data = AtomData(memory_session, selected_atoms="He, Be, B, N VI", chianti_ions="He 1; N 5; Si 1")
def test_atom_data_init(memory_session): nist = DataSource.as_unique(memory_session, short_name="nist-asd") ch = DataSource.as_unique(memory_session, short_name="chianti_v8.0.2") ku = DataSource.as_unique(memory_session, short_name="ku_latest") atom_data = AtomData(memory_session, selected_atoms="He, Be, B, N", chianti_ions="He 1; N 5") assert set(atom_data.selected_atomic_numbers) == set([2, 4, 5, 7]) assert set(atom_data.chianti_ions) == set([(2, 1), (7, 5)])
def atom_data_be(test_session): atom_data = AtomData(test_session, selected_atoms="Be") return atom_data
def atom_data(test_session, chianti_short_name): atom_data = AtomData(test_session, selected_atoms="He, Be, B, N, Si, Zn", chianti_ions="He 1; N 5", chianti_short_name=chianti_short_name) return atom_data
def atom_data(test_session): atom_data = AtomData(test_session, selected_atoms="He, Be, B, N, Si, Zn", chianti_ions="He 1; N 5") return atom_data