def test_should_return_1_between_alpha_one_alpha_two(self): scaler = restraints.PlateauSmoothScaler(0.2, 0.4, 0.6, 0.8, 1.0, 0.0) self.assertAlmostEqual(scaler(0.5), 1.0)
def test_should_return_correct_value_middle_down2(self): scaler = restraints.PlateauSmoothScaler(0.2, 0.4, 0.6, 0.8, 1.0, 0.0) self.assertAlmostEqual(scaler(0.25), 0.15625)
def test_should_raise_if_alpha_is_above_one(self): scaler = restraints.PlateauSmoothScaler(0.2, 0.5, 0.6, 0.8) with self.assertRaises(RuntimeError): scaler(2)
def test_should_return_0_above_alpha_max(self): scaler = restraints.PlateauSmoothScaler(0.2, 0.5, 0.6, 0.8, 1.0, 0.0) self.assertAlmostEqual(scaler(0.9), 0.0)
def test_should_raise_if_alpha_max_less_than_alpha_min(self): with self.assertRaises(RuntimeError): restraints.PlateauSmoothScaler(0.7, 0.5, 0.6, 0.6)
def test_should_raise_when_alpha_max_above_one(self): with self.assertRaises(RuntimeError): restraints.PlateauSmoothScaler(1, 0.5, 0.6, 2)
def test_should_raise_when_alpha_max_below_zero(self): with self.assertRaises(RuntimeError): restraints.PlateauSmoothScaler(1, 0.5, 0.6, -1)
def test_should_return_correct_value_middle_up2(self): scaler = restraints.PlateauSmoothScaler(0.2, 0.4, 0.6, 0.8) self.assertAlmostEqual(scaler(0.65), 0.84375)
def test_should_return_0_below_alpha_min(self): scaler = restraints.PlateauSmoothScaler(0.2, 0.5, 0.6, 0.8) self.assertAlmostEqual(scaler(0.1), 0.0)