def test_armstrong(self): self.assertTrue(is_armstrong(6)) self.assertTrue(is_armstrong(153)) self.assertFalse(is_armstrong(351))
def test_armstrong(number, expected): assert is_armstrong(number) == expected