an.tsouchlos/ba-thesis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Fixed proximal unit tests

Browse Source
This commit is contained in:
Andreas Tsouchlos 2022-11-25 11:30:55 +01:00
parent 8469e50a19
commit c9a4b1b61e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sw/test/test_proximal.py
View File

@ -20,7 +20,7 @@ class CheckParityTestCase(unittest.TestCase):
[0, 0, 0, 0, 0, 1, 0], [0, 0, 0, 0, 0, 1, 0],
[0, 0, 0, 0, 0, 0, 1]]) [0, 0, 0, 0, 0, 0, 1]])
decoder = proximal.ProximalDecoder(H, R) decoder = proximal.ProximalDecoder(H)
d1 = np.array([0, 1, 0, 1]) d1 = np.array([0, 1, 0, 1])
c1 = np.dot(np.transpose(G), d1) % 2 c1 = np.dot(np.transpose(G), d1) % 2
@ -62,7 +62,7 @@ class GradientTestCase(unittest.TestCase):
expected_grad_h = np.array( expected_grad_h = np.array(
[4, 26, -8, -36, 38, 28, -32]) # Manually calculated result [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) grad_h = decoder._grad_h(x)
self.assertEqual(np.array_equal(grad_h, expected_grad_h), True) self.assertEqual(np.array_equal(grad_h, expected_grad_h), True)