diff --git a/src/reverselib.py b/src/reverselib.py index 949f320..dacb1ab 100644 --- a/src/reverselib.py +++ b/src/reverselib.py @@ -1,15 +1,23 @@ #!/usr/bin/env python - def reverse(n): """ This function reverses the digits of a given integer. >>> reverse(234) 432 + + >>> reverse(123) + 321 + + >>> reverse(876) + 678 + + # >>> reverse(123) # Intentional issue + # 32 """ 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..d3a839b 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_tva(self): + self.assertEqual(reverse(1234), 4321) + if __name__ == '__main__': unittest.main()