diff --git a/functions/palindrome_number.py b/functions/palindrome_number.py index 26529ec..8382eba 100644 --- a/functions/palindrome_number.py +++ b/functions/palindrome_number.py @@ -1,2 +1,12 @@ def palindrome_number(x): - pass + temp = x + reverse = 0 + while(x>0): + dig = x%10 + reverse = reverse*10 + dig + x = x//10 + + if temp == reverse: + return True + else: + return False diff --git a/tests/palindrome_number_test.py b/tests/palindrome_number_test.py index 2603549..669271b 100644 --- a/tests/palindrome_number_test.py +++ b/tests/palindrome_number_test.py @@ -1,2 +1,34 @@ import pytest from functions.palindrome_number import palindrome_number + +def test_returns_true_for_99(): + # Arrange + n = 99 + + # Act + answer = palindrome_number(99) + + # Assert + assert answer == True + + +def test_returns_true_for_151(): + # Arrange + n = 151 + + # Act + answer = palindrome_number(n) + + # Assert + assert answer == True + + +def test_returns_false_for_123(): + # Arrange + n = 123 + + # Act + answer = palindrome_number(123) + + # Assert + assert answer == False