16 lines
460 B
Python
16 lines
460 B
Python
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
|