Python Tree.mean Examples

Example #1

File: test_main.py Project: nikitagrygoriev/Tree

 def test_mean(self):
     self.assertAlmostEqual(Tree.mean(root), 3.8)
     self.assertAlmostEqual(Tree.mean(root.left), 10 / 3)
     self.assertAlmostEqual(Tree.mean(root.right.left), 1)
     self.assertAlmostEqual(Tree.mean(root.right.right), 3.75)

Example #2

File: test_main.py Project: nikitagrygoriev/Tree

import unittest
from main import Tree

root = Tree(5)
root.insert(3, right=False)
root.left.insert(2, right=False)
root.left.insert(5)
root.insert(7)
root.right.insert(0)
root.right.insert(1, right=False)
root.right.right.insert(2, right=False)
root.right.right.insert(8)
root.right.right.right.insert(5)

print('Sum of the tree elements = ', root.sum())
print('Mean of the tree elements = ', root.mean())
print('Median of the tree elements = ', root.median())


class TestTree(unittest.TestCase):
    def test_sum(self):
        self.assertAlmostEqual(Tree.sum(root), 38)
        self.assertAlmostEqual(Tree.sum(root.left), 10)
        self.assertAlmostEqual(Tree.sum(root.right), 23)
        self.assertAlmostEqual(Tree.sum(root.left.left), 2)

    def test_mean(self):
        self.assertAlmostEqual(Tree.mean(root), 3.8)
        self.assertAlmostEqual(Tree.mean(root.left), 10 / 3)
        self.assertAlmostEqual(Tree.mean(root.right.left), 1)
        self.assertAlmostEqual(Tree.mean(root.right.right), 3.75)