Fix kernel regression test

ptycho · Jan 6, 2025 · 60c7e60 · 60c7e60
1 parent efe000f
commit 60c7e60
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/test/accelerate_tests/base_tests/gradient_descent_kernel_test.py b/test/accelerate_tests/base_tests/gradient_descent_kernel_test.py
@@ -168,22 +168,21 @@ def test_make_a012_notb(self):
         [ 0.,  0.,  0.]]], dtype=FLOAT_TYPE)
         np.testing.assert_array_almost_equal(exp_A1, GDK.npy.LLerr,
                                       err_msg="`LLerr` buffer (=A1) has not been updated as expected")
-        exp_A2 = np.array([[[ 0.,  4., 12.],
-        [ 4.,  8., 16.],
-        [12., 16., 24.]],
+        exp_A2 = np.array([[[ 0.,  2.,  8.],
+        [ 2.,  4., 10.],
+        [ 8., 10., 16.]],
 
-       [[ 0., 12., 28.],
-        [12., 24., 40.],
-        [28., 40., 56.]],
+       [[ 0.,  2.,  8.],
+        [ 2.,  4., 10.],
+        [ 8., 10., 16.]],
 
-       [[ 0., 20., 44.],
-        [20., 40., 64.],
-        [44., 64., 88.]],
+       [[ 0.,  2.,  8.],
+        [ 2.,  4., 10.],
+        [ 8., 10., 16.]],
 
        [[ 0.,  0.,  0.],
         [ 0.,  0.,  0.],
         [ 0.,  0.,  0.]]], dtype=FLOAT_TYPE)
-        print(GDK.npy.LLden)
         np.testing.assert_array_almost_equal(exp_A2, GDK.npy.LLden,
                                       err_msg="`LLden` buffer (=A2) has not been updated as expected")