def test_verify_problem(self):
"""
Test that verify only passes if all required values are set.
"""
fitting_problem = FittingProblem(self.options)
with self.assertRaises(exceptions.FittingProblemError):
fitting_problem.verify()
self.fail('verify() passes when no values are set.')
fitting_problem.starting_values = [OrderedDict([('p1', 1), ('p2', 2)])]
with self.assertRaises(exceptions.FittingProblemError):
fitting_problem.verify()
self.fail('verify() passes starting values are set.')
fitting_problem.data_x = np.array([1, 2, 3, 4, 5])
with self.assertRaises(exceptions.FittingProblemError):
fitting_problem.verify()
self.fail('verify() passes when data_x is set.')
fitting_problem.data_y = np.array([1, 2, 3, 4, 5])
with self.assertRaises(exceptions.FittingProblemError):
fitting_problem.verify()
self.fail('verify() passes when data_y is set.')
fitting_problem.function = lambda x, p1, p2: p1 + p2
try:
fitting_problem.verify()
except exceptions.FittingProblemError:
self.fail('verify() fails when all required values set.')
fitting_problem.data_x = [1, 2, 3]
with self.assertRaises(exceptions.FittingProblemError):
fitting_problem.verify()
self.fail('verify() passes for x values not numpy.')
def test_verify_problem(self):
"""
Test that verify only passes if all required values are set.
"""
fitting_problem = FittingProblem()
with self.assertRaises(TypeError):
fitting_problem.verify()
self.fail('verify() passes when no values are set.')
fitting_problem.starting_values = [['p1', [1]], ['p2', [2]]]
with self.assertRaises(TypeError):
fitting_problem.verify()
self.fail('verify() passes starting values are set.')
fitting_problem.data_x = np.array([1, 2, 3, 4, 5])
with self.assertRaises(TypeError):
fitting_problem.verify()
self.fail('verify() passes when data_x is set.')
fitting_problem.data_y = np.array([1, 2, 3, 4, 5])
with self.assertRaises(TypeError):
fitting_problem.verify()
self.fail('verify() passes when data_y is set.')
fitting_problem.functions = [[lambda x, p1, p2: p1 + p2, [1, 2]]]
try:
fitting_problem.verify()
except TypeError:
self.fail('verify() fails when all required values set.')
fitting_problem.data_x = [1, 2, 3]
with self.assertRaises(TypeError):
fitting_problem.verify()
self.fail('verify() passes for x values not numpy.')
