def test_addChild( self ): node1 = DagNode( 11 ) node2 = DagNode( 22 ) node1.addChild( node2 ) self.assertTrue( node2 in node1.getChildren() ) node2.addChild( node1 ) self.assertTrue( node1 in node2.getChildren() )
def test_addChild(self): node1 = DagNode(11) node2 = DagNode(22) node1.addChild(node2) self.assertTrue(node2 in node1.getChildren()) node2.addChild(node1) self.assertTrue(node1 in node2.getChildren())
def test_isRoot(self): node1 = DagNode(11) node2 = DagNode(22) node3 = DagNode(33) self.assertTrue(node1.isRoot()) node1.addChild(node2) self.assertTrue(node1.isRoot()) node3.addChild(node1) self.assertTrue(node1.isRoot()) node1.addParent(node3) self.assertFalse(node1.isRoot())
def test_isLeaf(self): node1 = DagNode(11) node2 = DagNode(22) node3 = DagNode(33) self.assertTrue(node1.isLeaf()) self.assertTrue(node2.isLeaf()) self.assertTrue(node3.isLeaf()) node1.addChild(node2) self.assertFalse(node1.isLeaf()) self.assertTrue(node2.isLeaf()) node3.addChild(node1) self.assertFalse(node1.isLeaf())
def test_isRoot( self ): node1 = DagNode( 11 ) node2 = DagNode( 22 ) node3 = DagNode( 33 ) self.assertTrue( node1.isRoot() ) node1.addChild( node2 ) self.assertTrue( node1.isRoot() ) node3.addChild( node1 ) self.assertTrue( node1.isRoot() ) node1.addParent( node3 ) self.assertFalse( node1.isRoot() )
def test_isLeaf( self ): node1 = DagNode( 11 ) node2 = DagNode( 22 ) node3 = DagNode( 33 ) self.assertTrue( node1.isLeaf() ) self.assertTrue( node2.isLeaf() ) self.assertTrue( node3.isLeaf() ) node1.addChild( node2 ) self.assertFalse( node1.isLeaf() ) self.assertTrue( node2.isLeaf() ) node3.addChild( node1 ) self.assertFalse( node1.isLeaf() )
def test_setColorRecursively(self): node1 = DagNode(11) node2 = DagNode(22) node3 = DagNode(33) node1.addChild(node2) node2.addChild(node3) self.assertEqual(node1.getColor(), DfsNode.White) self.assertEqual(node2.getColor(), DfsNode.White) self.assertEqual(node3.getColor(), DfsNode.White) node1.setColorRecursively(DfsNode.Black) self.assertEqual(node1.getColor(), DfsNode.Black) self.assertEqual(node2.getColor(), DfsNode.Black) self.assertEqual(node3.getColor(), DfsNode.Black)
def test_setColorRecursively( self ): node1 = DagNode( 11 ) node2 = DagNode( 22 ) node3 = DagNode( 33 ) node1.addChild( node2 ); node2.addChild( node3 ); self.assertEqual( node1.getColor(), DfsNode.White ) self.assertEqual( node2.getColor(), DfsNode.White ) self.assertEqual( node3.getColor(), DfsNode.White ) node1.setColorRecursively( DfsNode.Black ) self.assertEqual( node1.getColor(), DfsNode.Black ) self.assertEqual( node2.getColor(), DfsNode.Black ) self.assertEqual( node3.getColor(), DfsNode.Black )