From ebff46e78ce4ecc442c43fe44d4049d69bc8cc0a Mon Sep 17 00:00:00 2001 From: epapoutsellis Date: Wed, 10 Apr 2019 11:11:39 +0100 Subject: wip for with and without operators, functions --- Wrappers/Python/ccpi/optimisation/functions/BlockFunction.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'Wrappers') diff --git a/Wrappers/Python/ccpi/optimisation/functions/BlockFunction.py b/Wrappers/Python/ccpi/optimisation/functions/BlockFunction.py index a74a215..14105b5 100644 --- a/Wrappers/Python/ccpi/optimisation/functions/BlockFunction.py +++ b/Wrappers/Python/ccpi/optimisation/functions/BlockFunction.py @@ -124,8 +124,14 @@ if __name__ == '__main__': res_out = B.range_geometry().allocate() f.proximal_conjugate( U, tau, out = res_out) - numpy.testing.assert_array_almost_equal(res_no_out[0].as_array(), \ - res_out[0].as_array(), decimal=4) + numpy.testing.assert_array_almost_equal(res_no_out[0][0].as_array(), \ + res_out[0][0].as_array(), decimal=4) + + numpy.testing.assert_array_almost_equal(res_no_out[0][1].as_array(), \ + res_out[0][1].as_array(), decimal=4) + + numpy.testing.assert_array_almost_equal(res_no_out[1].as_array(), \ + res_out[1].as_array(), decimal=4) -- cgit v1.2.3