import pytest

from armstrong import is_armstrong


@pytest.mark.parametrize(
    "number, expected",
    [
        (5, True),
        (153, True),
        (370, True),
        (371, True),
        (4150, False),
        (2020, False),
        (9474, True),
        (1989, False),
        (11, False),
    ],
)
def test_armstrong(number, expected):
    assert is_armstrong(number) == expected

import pytest

from armstrong import is_armstrong


@pytest.mark.parametrize(
    "number, expected",
    [
        (5, True),
        (153, True),
        (370, True),
        (371, True),
        (4150, False),
        (2020, False),
        (9474, True),
        (1989, False),
        (11, False),
    ],
)
def test_armstrong(number, expected):
    assert is_armstrong(number) == expected
