kse-02/tests/test_check_armstrong.py

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=220) == False

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

    def test_check_armstrong_3(self):
        assert check_armstrong(n=35) == False

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