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

Fixed gradient unit test

Browse Source
This commit is contained in:
Andreas Tsouchlos 2022-11-04 21:35:20 +01:00
parent bf469afbd1
commit 01bc41b8c5
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

@ -37,9 +37,9 @@ class CheckParityTestCase(unittest.TestCase):
class GradientTestCase(unittest.TestCase): class GradientTestCase(unittest.TestCase):
"""Test case for the calculation of the gradient of the code-constraint-polynomial""" """Test case for the calculation of the gradient of the code-constraint-polynomial"""
def test_grad_h(self): def test_grad_h(self):
H = np.array([[1, 0, 1], H = np.array([[1, 0, 0],
[0, 1, 0]]) [0, 1, 0]])
x = np.array([2, 3, 4]) x = np.array([1, 2, 2])
decoder = proximal.ProximalDecoder(H) decoder = proximal.ProximalDecoder(H)
grad = decoder._grad_h(x) grad = decoder._grad_h(x)