def test_create_row_from_child_n_parents(self): child = BNNode([0,1], [0,0,1]) p1 = BNNode([0,1], [1,0,1]) p2 = BNNode([0,1], [0,1,0]) parents = [p1,p2] row1 = BN.create_row_from_child_n_parents(child, parents, 0) self.assertEquals(True, BN.lists_match([1,0,0], row1)) row2 = BN.create_row_from_child_n_parents(child, parents, 1) self.assertEquals(True, BN.lists_match([0,1,0], row2)) row3 = BN.create_row_from_child_n_parents(child, parents, 2) self.assertEquals(True, BN.lists_match([1,0,1], row3))
def test_lists_match(self): l1 = [0,1,2,3,4] l2 = [0,1,2,3,4] self.assertEquals(True, BN.lists_match([0,1,2,3,4], [0,1,2,3,4])) self.assertEquals(False, BN.lists_match([1,2,3], [1])) self.assertEquals(False, BN.lists_match([1,2,3], [1,2,4]))