diff --git a/.gitignore b/.gitignore index bee8a64..91a5af7 100644 --- a/.gitignore +++ b/.gitignore @@ -1 +1,7 @@ __pycache__ +.vs/ +.vscode/ +__pycache__/ +*.pyc +venv/ +.env \ No newline at end of file diff --git a/src/reverselib.py b/src/reverselib.py index 949f320..351fc75 100644 --- a/src/reverselib.py +++ b/src/reverselib.py @@ -7,9 +7,12 @@ def reverse(n): >>> reverse(234) 432 + + >>> reverse(12345) + 54321 """ 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..3788340 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_another_one(self): + self.assertEqual(reverse(100), 1) + if __name__ == '__main__': unittest.main()