From 01bc41b8c51f98292184e46551e196399c664c5e Mon Sep 17 00:00:00 2001 From: Andreas Tsouchlos Date: Fri, 4 Nov 2022 21:35:20 +0100 Subject: [PATCH] Fixed gradient unit test --- sw/test/test_proximal.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sw/test/test_proximal.py b/sw/test/test_proximal.py index ff49684..ed3afb8 100644 --- a/sw/test/test_proximal.py +++ b/sw/test/test_proximal.py @@ -37,9 +37,9 @@ class CheckParityTestCase(unittest.TestCase): class GradientTestCase(unittest.TestCase): """Test case for the calculation of the gradient of the code-constraint-polynomial""" def test_grad_h(self): - H = np.array([[1, 0, 1], + H = np.array([[1, 0, 0], [0, 1, 0]]) - x = np.array([2, 3, 4]) + x = np.array([1, 2, 2]) decoder = proximal.ProximalDecoder(H) grad = decoder._grad_h(x)