def test_set_with_config_item_string_item_access():
config_ns = ConfigurationNameSpace(simple_config_dict)
config_ns.set_config_item('a.b.param2.item0', 2)
item = config_ns.get_config_item('a.b.param2.item0')
assert item == 2
def test_set_with_config_item_string_item_access_quantity():
config_ns = ConfigurationNameSpace(simple_config_dict)
config_ns.set_config_item('a.b.param2.item2', 7)
item = config_ns.get_config_item('a.b.param2.item2')
assert_almost_equal(item, 7 * u.km)
def test_simple_configuration_namespace():
config_ns = ConfigurationNameSpace(simple_config_dict)
assert config_ns.a.b.param1 == 1
config_ns.a.b.param1 = 2
assert (config_ns['a']['b']['param1'] == 2)
config_ns['a']['b']['param1'] = 3
assert config_ns.a.b.param1 == 3
def test_quantity_configuration_namespace():
config_ns = ConfigurationNameSpace(simple_config_dict)
config_ns.a.b.param3 = 3
assert_almost_equal(config_ns['a']['b']['param3'], 3 * u.km)
config_ns.a.b.param3 = 5000 * u.m
assert_almost_equal(config_ns['a']['b']['param3'], 5 * u.km)
def test_quantity_configuration_namespace():
config_ns = ConfigurationNameSpace(simple_config_dict)
config_ns.a.b.param3 = 3
assert_almost_equal(config_ns["a"]["b"]["param3"].to(u.km).value, 3)
config_ns.a.b.param3 = 5000 * u.m
assert_almost_equal(config_ns["a"]["b"]["param3"].to(u.km).value, 5)
def test_simple_configuration_namespace():
config_ns = ConfigurationNameSpace(simple_config_dict)
assert config_ns.a.b.param1 == 1
config_ns.a.b.param1 = 2
assert config_ns["a"]["b"]["param1"] == 2
config_ns["a"]["b"]["param1"] = 3
assert config_ns.a.b.param1 == 3
def setup(self):
self.nlte_species = [(2, 0), (2, 1)]
self.nlte_config = ConfigurationNameSpace(
{'species': self.nlte_species})
self.atom_data = atomic.AtomData.from_hdf5(helium_test_db)
self.plasma = plasma_array.BasePlasmaArray.from_abundance(
{'He': 1.0},
1e-15 * u.Unit('g/cm3'),
self.atom_data,
10 * u.day,
nlte_config=self.nlte_config)
self.plasma.j_blues = 0.5 * pd.DataFrame(
intensity_black_body(self.atom_data.lines.nu.values[np.newaxis].T,
np.array([10000.])))
self.plasma.tau_sobolevs = pd.DataFrame(
np.zeros_like(self.plasma.j_blues))
self.plasma.t_rads = np.array([10000.])
self.plasma.t_electrons = np.array([9000.])
self.plasma.ws = np.array([0.5])
self.plasma.electron_densities = pd.Series([1.e9])
self.plasma.ion_populations = ion_populations
self.plasma.calculate_nlte_level_populations()
def test_config_namespace_copy():
config_ns = ConfigurationNameSpace(simple_config_dict)
config_ns2 = config_ns.deepcopy()
config_ns2.a.b.param1 = 2
assert config_ns2.a.b.param1 != config_ns.a.b.param1
def test_set_with_config_item_string_quantity():
config_ns = ConfigurationNameSpace(simple_config_dict)
config_ns.set_config_item('a.b.param3', 2)
assert_almost_equal(config_ns.a.b.param3, 2 * u.km)
def test_access_with_config_item_string():
config_ns = ConfigurationNameSpace(simple_config_dict)
assert config_ns.get_config_item('a.b.param1') == 1
config_ns.set_config_item('a.b.param1', 2)
assert config_ns.a.b.param1 == 2
def config_ns():
"""Return example instance of `ConfigurationNameSpace` class."""
config_name_space = ConfigurationNameSpace(simple_config_dict)
return config_name_space
def test_get_with_config_item_string_item_access():
config_ns = ConfigurationNameSpace(simple_config_dict)
item = config_ns.get_config_item("a.b.param2.item0")
assert item == 0
item = config_ns.get_config_item("a.b.param2.item1")
assert item == 1
