def test_createStub(self):
n1 = Node(10, 10)
stub = n1.createStub(5)
self.assertIsNotNone(stub)
self.assertEqual(stub.width, 5)
self.assertEqual(n1.idealPos, stub.idealPos)
self.assertEqual(n1.data, stub.data)
def test_createStub(self):
n1 = Node(10, 10)
stub = n1.createStub(5)
self.assertIsNotNone(stub)
self.assertEqual(stub.width, 5)
self.assertEqual(n1.idealPos, stub.idealPos)
self.assertEqual(n1.data, stub.data)
def test_getRoot(self):
n1 = Node(10, 10)
self.assertEqual(n1.getRoot(), n1)
n2 = n1.createStub(5)
self.assertEqual(n1.getRoot(), n2)
n3 = n2.createStub(5)
self.assertEqual(n1.getRoot(), n3)
def test_getRoot(self):
n1 = Node(10, 10)
self.assertEqual(n1.getRoot(), n1)
n2 = n1.createStub(5)
self.assertEqual(n1.getRoot(), n2)
n3 = n2.createStub(5)
self.assertEqual(n1.getRoot(), n3)
def test_pathLength(self):
n1 = Node(1, 50)
n1.currentPos = 20
n2 = Node(2, 50)
n3 = Node(804, 50)
stub = n3.createStub()
n3.currentPos = 810
n4 = Node(854, 50)
n4.currentPos = 800
stub4 = n4.createStub()
stub4.currentPos = 700
stub4_2 = stub4.createStub()
# should return 0 if the input is empty
self.assertEqual(metrics.pathLength([]), 0)
# should return sum of the displacements from leaves to stubs up to
# root
self.assertEqual(metrics.pathLength([n1]), 19)
self.assertEqual(metrics.pathLength([n2]), 0)
self.assertEqual(metrics.pathLength([n3]), 6)
self.assertEqual(metrics.pathLength([n4, stub4, stub4_2]), 254)
self.assertEqual(
metrics.pathLength([[n1, n2, n3, stub4_2], [stub4], [n4]]),
279 / 4)
def test_removeStub(self):
n1 = Node(10, 10)
stub = n1.createStub(5)
n1.removeStub()
self.assertIsNone(n1.parent)
self.assertIsNone(stub.child)
def test_getPathToRootLength(self):
n4 = Node(854, 50)
n4.currentPos = 800
stub4 = n4.createStub()
stub4.currentPos = 700
self.assertEqual(n4.getPathToRootLength(), 254)
def test_getPathFromRoot(self):
n1 = Node(10, 10)
self.assertEqual(n1.getPathFromRoot(), [n1])
n2 = n1.createStub(5)
n3 = n2.createStub(5)
self.assertEqual(n1.getPathFromRoot(), [n3, n2, n1])
def test_isStub(self):
n1 = Node(10, 10)
stub = n1.createStub(5)
self.assertFalse(n1.isStub())
self.assertTrue(stub.isStub())
def test_removeStub(self):
n1 = Node(10, 10)
stub = n1.createStub(5)
n1.removeStub()
self.assertIsNone(n1.parent)
self.assertIsNone(stub.child)
def test_getPathToRootLength(self):
n4 = Node(854, 50)
n4.currentPos = 800
stub4 = n4.createStub()
stub4.currentPos = 700
self.assertEqual(n4.getPathToRootLength(), 254)
def test_getPathFromRoot(self):
n1 = Node(10, 10)
self.assertEqual(n1.getPathFromRoot(), [n1])
n2 = n1.createStub(5)
n3 = n2.createStub(5)
self.assertEqual(n1.getPathFromRoot(), [n3, n2, n1])
def test_isStub(self):
n1 = Node(10, 10)
stub = n1.createStub(5)
self.assertFalse(n1.isStub())
self.assertTrue(stub.isStub())
