From c9a4b1b61ee0fdddc0eeea21a5d0cab56a509a84 Mon Sep 17 00:00:00 2001 From: Andreas Tsouchlos Date: Fri, 25 Nov 2022 11:30:55 +0100 Subject: [PATCH] Fixed proximal unit tests --- 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 355efc6..650b0e6 100644 --- a/sw/test/test_proximal.py +++ b/sw/test/test_proximal.py @@ -20,7 +20,7 @@ class CheckParityTestCase(unittest.TestCase): [0, 0, 0, 0, 0, 1, 0], [0, 0, 0, 0, 0, 0, 1]]) - decoder = proximal.ProximalDecoder(H, R) + decoder = proximal.ProximalDecoder(H) d1 = np.array([0, 1, 0, 1]) c1 = np.dot(np.transpose(G), d1) % 2 @@ -62,7 +62,7 @@ class GradientTestCase(unittest.TestCase): expected_grad_h = np.array( [4, 26, -8, -36, 38, 28, -32]) # Manually calculated result - decoder = proximal.ProximalDecoder(H, R) + decoder = proximal.ProximalDecoder(H) grad_h = decoder._grad_h(x) self.assertEqual(np.array_equal(grad_h, expected_grad_h), True)