diff --git a/tests/model/test_model.py b/tests/model/test_model.py index 8eaf597..6f3ef8f 100644 --- a/tests/model/test_model.py +++ b/tests/model/test_model.py @@ -14,6 +14,27 @@ def test_dotdipolemodel_m(): assert mod.m == 2 +def test_dotdipolemodel_cost(): + mod = model.DotDipoleModel([model.DotMeasurement(1, (0, 0, 1)), model.DotMeasurement(2, (1, 0, 0))], 1) + costs = mod.costs() + jac = mod.jac() + + pt_to_test = numpy.array((1, 2, 3, 4, 5, 6)) + expected_cost = [-1.05408565512728256, -2.05293155269909457] + expected_jacobian = [ + [ + -0.007460090362383803, -0.009325112952979752, -0.009325112952979752, + 0.007968732887091788, 0.00856214916591777, 0.006697126575321822 + ], [ + -0.00512240832571883, -0.008537347209531383, -0.01024481665143766, + 0.005098015905120168, 0.007927536694564856, 0.008488562368334061 + ] + ] + + numpy.testing.assert_allclose(costs(pt_to_test), expected_cost, err_msg="Costs aren't as expected.") + numpy.testing.assert_allclose(jac(pt_to_test), expected_jacobian, err_msg="Jacobian aren't as expected.") + + def print_result(msg, result): print(msg) print(f"\tResult: {result.x}") diff --git a/tests/test_pathfinder.py b/tests/test_pathfinder.py index 2fa270f..86c14dc 100644 --- a/tests/test_pathfinder.py +++ b/tests/test_pathfinder.py @@ -1,5 +1,6 @@ from pathfinder import __version__ - +import pathfinder def test_version(): assert __version__ == '0.0.1' + assert pathfinder.get_version() == __version__ \ No newline at end of file