From 43cf9c29d483cec45283a0bec6ad36b1c13ce10e Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Tue, 10 Jan 2023 10:27:56 +0000 Subject: [PATCH] test: test for multi parameter foot feature added. --- tests/roots_test.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/roots_test.py b/tests/roots_test.py index b6e9bdb0..89feba96 100644 --- a/tests/roots_test.py +++ b/tests/roots_test.py @@ -42,3 +42,13 @@ def test_root_no_autograd(): with pytest.raises(Exception): my_root = pe.roots.find_root(my_obs, root_function) + + +def test_root_multi_parameter(): + o1 = pe.pseudo_Obs(1.1, 0.1, "test") + o2 = pe.pseudo_Obs(1.3, 0.12, "test") + + f2 = lambda x, d: d[0] + d[1] * x + + assert f2(-o1 / o2, [o1, o2]) == 0 + assert pe.find_root([o1, o2], f2) == -o1 / o2