A few tests to for the is_narcissistic() function

test_narcis.py

import unittest

import solution


class TestIsNarcissistic(unittest.TestCase):

    def test_is_narcissistic(self):

        self.assertFalse(
            solution.is_narcissistic('999'))

        self.assertTrue(
            solution.is_narcissistic('0'))

        self.assertTrue(
            solution.is_narcissistic('407'))

        self.assertFalse(
            solution.is_narcissistic('470'))

        self.assertFalse(
            solution.is_narcissistic('ZZF02', 36))

        self.assertTrue(
            solution.is_narcissistic('223', 4))

        self.assertTrue(
            solution.is_narcissistic('115132219018763992565095597973971522400'))

        self.assertTrue(
            solution.is_narcissistic('122', 3))

        self.assertTrue(
            solution.is_narcissistic('124031', 5))

        self.assertTrue(
            solution.is_narcissistic('12205', 7))

        self.assertTrue(
            solution.is_narcissistic('45', 9))

        self.assertTrue(
            solution.is_narcissistic('12344AA12A721803422912A8AA4963568083A268456A4', 11))

        self.assertTrue(
            solution.is_narcissistic('15079346A6B3B14BB56B395898B96629A8B01515344B4B0714B', 12))


if __name__ == '__main__':
    unittest.main()

Благодаря за тестовете! :)