def setUp(self):
self.ab_cd = Tree(Tree(Leaf("a"), Leaf("b")),
Tree(Leaf("c"), Leaf("d")))
self.ab = self.ab_cd.left
self.cd = self.ab_cd.right
self.a = self.ab.left
self.b = self.ab.right
self.c = self.cd.left
self.d = self.cd.right
pass
def test_tree_contains_all_subtrees(self):
ab_c = Tree(self.ab, self.c)
self.assertTrue(ab_c.contains(ab_c.left))
self.assertTrue(ab_c.contains(ab_c.right))
def test_trees_with_switched_branches_are_equal(self):
cd_ba = Tree(self.cd, Tree(self.b, self.a))
self.assertEqual(self.ab_cd, cd_ba)
def test_unequal_trees_are_unequal(self):
aa = Tree(self.a, self.a)
self.assertNotEqual(aa, self.ab)
