def test_busca_não_encontra(self):
a = ABB()
a.adicionar(5)
a.adicionar(3)
a.adicionar(4)
a.adicionar(2)
a.adicionar(7)
a.adicionar(8)
a.adicionar(9)
self.assertIsNone(a.busca(1))
self.assertIsNone(a.busca(6))
self.assertIsNone(a.busca(10))
def test_busca_encontra(self):
a = ABB()
a.adicionar(5)
a.adicionar(3)
a.adicionar(4)
a.adicionar(2)
a.adicionar(7)
a.adicionar(8)
a.adicionar(9)
self.assertTrue(a.busca(2).chave, 2)
self.assertTrue(a.busca(3).chave, 3)
self.assertTrue(a.busca(4).chave, 4)
self.assertTrue(a.busca(7).chave, 7)
self.assertTrue(a.busca(8).chave, 8)
self.assertTrue(a.busca(9).chave, 9)
def test_busca_encontra_raiz(self):
a = ABB()
a.adicionar(5)
a.adicionar(3)
a.adicionar(4)
a.adicionar(2)
a.adicionar(7)
a.adicionar(8)
a.adicionar(9)
self.assertTrue(a.busca(5).chave, 5)