38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
# These tests are auto-generated with test data from:
|
|
# https://github.com/exercism/problem-specifications/tree/main/exercises/armstrong-numbers/canonical-data.json
|
|
# File last updated on 2023-07-20
|
|
|
|
import unittest
|
|
|
|
from armstrong_numbers import (
|
|
is_armstrong_number,
|
|
)
|
|
|
|
|
|
class ArmstrongNumbersTest(unittest.TestCase):
|
|
def test_zero_is_an_armstrong_number(self):
|
|
self.assertIs(is_armstrong_number(0), True)
|
|
|
|
def test_single_digit_numbers_are_armstrong_numbers(self):
|
|
self.assertIs(is_armstrong_number(5), True)
|
|
|
|
def test_there_are_no_two_digit_armstrong_numbers(self):
|
|
self.assertIs(is_armstrong_number(10), False)
|
|
|
|
def test_three_digit_number_that_is_an_armstrong_number(self):
|
|
self.assertIs(is_armstrong_number(153), True)
|
|
|
|
def test_three_digit_number_that_is_not_an_armstrong_number(self):
|
|
self.assertIs(is_armstrong_number(100), False)
|
|
|
|
def test_four_digit_number_that_is_an_armstrong_number(self):
|
|
self.assertIs(is_armstrong_number(9474), True)
|
|
|
|
def test_four_digit_number_that_is_not_an_armstrong_number(self):
|
|
self.assertIs(is_armstrong_number(9475), False)
|
|
|
|
def test_seven_digit_number_that_is_an_armstrong_number(self):
|
|
self.assertIs(is_armstrong_number(9926315), True)
|
|
|
|
def test_seven_digit_number_that_is_not_an_armstrong_number(self):
|
|
self.assertIs(is_armstrong_number(9926314), False)
|