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 )
