20 lines
544 B
Python
20 lines
544 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=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
|