def test_constructor_with_delta(): p = Parameter('test_parameter', 1.0, delta=0.3) assert p.delta == 0.3 p.display()
def test_set_units(): p = Parameter('test_parameter', 1.0, unit=u.keV) p.value = 3.0 * u.MeV assert p.value == 3000.0 with pytest.raises(u.UnitConversionError): p.value = 3.0 * u.cm with pytest.raises(CannotConvertValueToNewUnits): p.unit = u.cm with pytest.raises(CannotConvertValueToNewUnits): p.unit = u.dimensionless_unscaled p.unit = u.MeV assert p.unit == u.MeV p.display()
def test_constructor_with_units(): p = Parameter('test_parameter', 1.0, unit=u.keV) assert p.unit == u.keV p.display()
def test_constructor_complete(): p = Parameter( 'test_parameter', 1.0, min_value=-5.0, max_value=5.0, delta=0.2, desc='test', free=False, unit=u.MeV, prior=Uniform_prior(), is_normalization=True, ) assert p.min_value == -5.0 assert p.max_value == 5.0 assert p.value == 1.0 assert p.delta == 0.2 assert p.name == 'test_parameter' assert p.description == 'test' assert p.fix == True assert p.free == False assert p.has_prior() == True assert p.unit == u.MeV assert p.is_normalization p.display()
def test_constructor_with_boundaries(): p = Parameter('test_parameter', 1.0, min_value=-5, max_value=5) assert p.min_value == -5 assert p.max_value == 5 p.display()
def test_set_within_bounds_units(): p = Parameter('test_parameter',1.0 * u.keV, min_value = -2.0 * u.MeV, max_value = 2.0 * u.MeV, unit=u.keV) p.value = 1.2 * u.MeV assert p.value == 1200.0 p.display()
def test_set_no_units(): p = Parameter('test_parameter', 1.0) p.value = 25.4 assert p.value == 25.4 p.display()
def test_set_delta_nounits(): p = Parameter('test_parameter', 1.0) p.delta = 0.5 assert p.delta == 0.5 p.display()
def test_set_delta_units(): p = Parameter('test_parameter', 1.0, unit='GeV') p.delta = 500 * u.MeV assert p.delta == 0.5 p.display()
def test_set_within_bounds_no_units(): p = Parameter('test_parameter', 1.0, min_value=-2.0, max_value=2.0) p.value = 1.5 assert p.value == 1.5 p.display()
def test_set_bounds_units(): p = Parameter('test_parameter', 1.0 * u.keV) p.bounds = (-2.0 * u.MeV, 2.0 * u.eV) assert p.min_value == -2000 assert p.max_value == 2e-3 p.display()
def test_duplicate(): p1 = Parameter('test_parameter', 1.0, min_value=-5.0, max_value=5.0, delta=0.2, desc='test', free=False, unit='MeV') p2 = p1.duplicate() assert p1.to_dict() == p2.to_dict() p1.display() p2.display()
def test_set_bounds_nounits(): p = Parameter('test_parameter', 1.0) p.bounds = (-2.0, 2.0) assert p.min_value == -2.0 assert p.max_value == 2.0 p.display()
def test_internal_setting(): p = Parameter('test_parameter', 1.0) p._set_internal_value(5.) p._get_internal_value() p.display() p._get_internal_min_value() p._get_internal_max_value()
def test_set_outside_bounds_units(): p = Parameter('test_parameter', 1.0 * u.keV, min_value = -2.0 * u.MeV, max_value = 2.0 * u.MeV, unit=u.keV) with pytest.raises(SettingOutOfBounds): p.value = -10.0 * u.MeV with pytest.raises(SettingOutOfBounds): p.value = 10.0 * u.MeV p.display()
def test_set_outside_bounds_no_units(): p = Parameter('test_parameter', 1.0, min_value=-2.0, max_value=2.0) with pytest.raises(SettingOutOfBounds): p.value = -10.0 with pytest.raises(SettingOutOfBounds): p.value = 10.0 p.display()
def test_set_bounds_nounits(): p = Parameter('test_parameter', 1.0) p.bounds = (-2.0 ,2.0) assert p.min_value == -2.0 assert p.max_value == 2.0 p.display() with pytest.warns(RuntimeWarning): p.value = 1.0 p.min_value = 2.0
def test_set_auxiliary_variable(): p1 = Parameter('test_parameter', 1.0, min_value=-5.0, max_value=5.0, delta=0.2, desc='test', free=False, unit='MeV') x = Parameter('aux_variable', 1.0) # ax + b law = Line() law.a = 2.0 law.b = 1.0 p1.add_auxiliary_variable(x, law) assert p1.has_auxiliary_variable() == True assert p1.value == 3.0 assert p1.free == False x.value = 4.0 assert p1.value == 6.0 # Check that assigning to the parameter doesn't produce any effect p1.value = -1.0 assert p1.value == 6.0 # Now check errors reporting with pytest.raises(AttributeError): p1.add_auxiliary_variable(1.0, law) # Now add it twice to verify that it overwrites it p1.add_auxiliary_variable(x, law) p1.add_auxiliary_variable(x, law) p1.display()
def test_conflicting_units_in_initial_value_and_unit_keyword(): p = Parameter('test_parameter', 1.0 * u.keV, desc='Description', unit=u.MeV) assert p.min_value is None assert p.max_value is None assert p.value == 1.0e-3 assert isinstance(p.delta, float) assert p.name == 'test_parameter' assert p.description == 'Description' assert p.fix == False assert p.free == True assert p.has_prior() == False assert p.prior is None assert p.unit == u.MeV p.display()
def test_default_constructor_units(): p = Parameter('test_parameter', 1.0 * u.keV, desc='Description') assert p.min_value is None assert p.max_value is None assert p.value == 1.0 assert isinstance(p.delta, float) assert p.name == 'test_parameter' assert p.description == 'Description' assert p.fix == False assert p.free == True assert p.has_prior() == False assert p.prior is None assert p.unit == u.keV p.display()
def test_default_constructor(): p = Parameter('test_parameter', 1.0, desc='Description') assert p.min_value is None assert p.max_value is None assert p.value == 1.0 assert isinstance(p.delta, float) assert p.name == 'test_parameter' assert p.description == 'Description' assert p.fix == False assert p.free == True assert p.has_prior() == False assert p.prior is None assert p.unit == u.dimensionless_unscaled # Test that we cannot call a parameter with a name with spaces in it with pytest.raises(AssertionError): _ = Parameter('test parameter 2', 1.0) # Test some failures cases with pytest.raises(ValueError): _ = Parameter('test', 'pippo') with pytest.raises(ValueError): _ = Parameter('test', 1.0, min_value='a') with pytest.raises(ValueError): _ = Parameter('test', 1.0, max_value='b') with pytest.raises(TypeError): _ = Parameter('test', 1.0, delta='b') p.display()