from unittest import TestCase
from benchmark.check_armstrong import check_armstrong


class Test_check_armstrong(TestCase):
    #  distances_true = {1: [583], 2: [582], 3: [433], 4: [0, 0, 0, 1], 5: [81]}
    #  distances_false = {1: [0], 2: [0], 3: [0], 4: [583, 58, 5, 0], 5: [0]}
    def test_check_armstrong_1(self):
        assert check_armstrong(n=583) == False

    #  distances_true = {1: [0]}
    #  distances_false = {1: [1]}
    def test_check_armstrong_2(self):
        assert check_armstrong(n=0) == True

    #  distances_true = {1: [153], 2: [152], 3: [3], 4: [0, 0, 0, 1], 5: [0]}
    #  distances_false = {1: [0], 2: [0], 3: [0], 4: [153, 15, 1, 0], 5: [1]}
    def test_check_armstrong_3(self):
        assert check_armstrong(n=153) == True

    #  distances_true = {1: [5], 2: [4], 3: [0]}
    #  distances_false = {1: [0], 2: [0], 3: [146]}
    def test_check_armstrong_4(self):
        assert check_armstrong(n=5) == False