diff --git a/src/reverselib.py b/src/reverselib.py index 949f320..a1d6e3c 100644 --- a/src/reverselib.py +++ b/src/reverselib.py @@ -7,9 +7,11 @@ def reverse(n): >>> reverse(234) 432 + >>> reverse(2026) + 6202 """ r = 0 - while n > 1: + while n > 0: r = r * 10 + n % 10 n = n // 10 return r diff --git a/tests/test_reverseno.py b/tests/test_reverseno.py index f75b70c..942983c 100644 --- a/tests/test_reverseno.py +++ b/tests/test_reverseno.py @@ -8,5 +8,8 @@ def test_zero(self): def test_ott(self): self.assertEqual(reverse(234), 432) + def test_jennifer(self): + self.assertEqual(reverse(2026), 6202) + if __name__ == '__main__': unittest.main()