def test1(self): numero_de_bolinas = 2 lista_de_bolinas = [[1, 1000], [1, 2000]] resultado = [2000, 1000] self.assertEqual(enfeites.solve(lista_de_bolinas), resultado)
def test_uma_bolinha(self): entrada = [[2, 10001]] saida = [10001] self.assertEqual(enfeites.solve(entrada), saida)
def test_last_ultimate(self): entrada = [[4, 98622], [9, 91], [9, 911], [5, 387], [9, 376], [10, 3], [1, 1505], [4, 14], [6, 102], [9, 105]] saida = [1505, 98622, 14, 387, 102, 911, 376, 105, 91, 3] self.assertEqual(enfeites.solve(entrada), saida)
def test_sete_bolinhas(self): entrada = [[2, 10001], [2, 90090], [4, 70007], [3, 44444], [2, 80008], [4, 50005], [3, 20022]] saida = [90090, 80008, 10001, 44444, 20022, 70007, 50005] self.assertEqual(enfeites.solve(entrada), saida)
def test_duas_caixinhas(self): entrada = [[2, 10001], [2, 90090], [7, 75655], [7, 10032], [2, 70005]] saida = [90090, 70005, 10001, 75655, 10032] self.assertEqual(enfeites.solve(entrada), saida)
def test_duas_bolinha(self): entrada = [[2, 10001], [2, 90090]] saida = [90090, 10001] self.assertEqual(enfeites.solve(entrada), saida)