palindrome.py

import unittest
from collections import defaultdict


def check_palindrome_permutation(input_string):
    """입력받은 문자열이 palindrome의 순열을 갖는지 확인한다."""
    char_count = defaultdict(int)

    for char in input_string:
        char_count[char] += 1

    odd_occurred = False
    for char, count in char_count.items():
        if count % 2 == 1:
            if odd_occurred:
                return False
            else:
                odd_occurred = True
    return True


def check_palindrome(input_string):
    """입력받은 문자열이 palindrome 인지 확인한다."""
    for i in range(len(input_string) // 2):
        if input_string[i] != input_string[-1 - i]:
            return False
    return True


class TestCheckPalindromePermutation(unittest.TestCase):
    def test_check_check_palindrome_permutation(self):
        self.assertTrue(check_palindrome_permutation('treret'))
        self.assertTrue(check_palindrome_permutation('akayk'))
        self.assertFalse(check_palindrome_permutation('levels'))


class TestCheckPalindrome(unittest.TestCase):
    def test_check_palindrome(self):
        self.assertTrue(check_palindrome('terret'))
        self.assertTrue(check_palindrome('kayak'))
        self.assertFalse(check_palindrome('levels'))