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