def test_rg_parity_spc_ones_spc_AF_1(self): node = GFastSSCNode(mask=self.rg_parity_ones_spc, AF=1) self.assertEqual(node.node_type, NodeTypes.RG_PARITY) expected_result = np.array([ 1, 1, 0, 1, 0, 1, 1, 0, 1, 1, 1, 0, 0, 1, 0, 1, ]) node.alpha = self.llr node() np.testing.assert_array_equal(node.beta, expected_result)
def test_g_repetition_spc(self): node = GFastSSCNode(mask=self.g_rep_spc) self.assertEqual(node.node_type, NodeTypes.G_REPETITION) expected_result = np.array([ 1, 1, 0, 0, 1, 1, 0, 0, 1, 1, 0, 0, 1, 1, 0, 0, ]) node.alpha = self.llr node() np.testing.assert_array_equal(node.beta, expected_result)
def test_rg_parity_spc_ones_spc_AF_3(self): mask = np.array([ 0, 0, 0, 0, 0, 1, 1, 1, 0, 1, 1, 1, 0, 1, 1, 1, ]) node = GFastSSCNode(mask=mask, AF=2) self.assertNotEqual(node.node_type, NodeTypes.RG_PARITY) node = GFastSSCNode(mask=mask, AF=3) self.assertEqual(node.node_type, NodeTypes.RG_PARITY)
def test_rg_parity_spc_ones_long_AF_1(self): node = GFastSSCNode(mask=self.rg_parity_ones_spc_long, AF=1) self.assertEqual(node.node_type, NodeTypes.RG_PARITY)
def test_rg_parity_ones_long(self): node = GFastSSCNode(mask=self.rg_parity_ones_long) self.assertEqual(node.node_type, NodeTypes.RG_PARITY)
def test_g_repetition_spc_long(self): node = GFastSSCNode(mask=self.g_rep_spc_long) self.assertEqual(node.node_type, NodeTypes.G_REPETITION)