From 62f203fcc30e52c6104047d3b2d55b611e1eacd1 Mon Sep 17 00:00:00 2001 From: Nicola Demo Date: Wed, 20 Jul 2022 17:59:01 +0200 Subject: [PATCH] Update test_operators.py --- tests/test_operators.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_operators.py b/tests/test_operators.py index 5f15e6b..6ddfd77 100644 --- a/tests/test_operators.py +++ b/tests/test_operators.py @@ -33,9 +33,9 @@ def test_grad_vector_output(): def test_div_vector_output(): grad_tensor_v = div(tensor_v, inp) - assert grad_tensor_v.shape == (20, 3) + assert grad_tensor_v.shape == (20, 1) grad_tensor_v = div(tensor_v, inp, components=['a', 'b'], d=['x', 'mu']) - assert grad_tensor_v.shape == (inp.shape[0], 2) + assert grad_tensor_v.shape == (inp.shape[0], 1) def test_nabla_scalar_output(): laplace_tensor_v = nabla(tensor_s, inp, components=['a'], d=['x', 'y'])