Fixed gradient unit test

2022-11-04 21:35:20 +01:00 · 2022-11-04 21:35:20 +01:00 · 01bc41b8c5
commit 01bc41b8c5
parent bf469afbd1
1 changed files with 2 additions and 2 deletions
--- 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)