from unittest import TestCase

from benchmark.check_armstrong import check_armstrong


class Test_check_armstrong(TestCase):
    def test_check_armstrong_1(self):
        assert check_armstrong(n=370) == True

    def test_check_armstrong_2(self):
        assert check_armstrong(n=5) == False

    def test_check_armstrong_3(self):
        assert check_armstrong(n=1) == True

    def test_check_armstrong_4(self):
        assert check_armstrong(n=0) == True

    def test_check_armstrong_5(self):
        assert check_armstrong(n=346) == False