add unit test and fix bug

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

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()