gistfile1.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

'''
ABCDEFGHIJKLMNOPQRSTUVWXYZ
QWERTYUIOPASDFGHJKLZXCVBNM?
に置換するプログラムを作る
A-Z 以外の文字は ? にする

サンプルの文字列は KSOIDHEPZ -> ALGORITHM
'''

import unittest


class TestSub(unittest.TestCase):

    def setUp(self):
        self.coded = 'KSOIDHEPZ'
        pass

    def testSub(self):
        self.assertEqual(simpledecode(self.coded), 'ALGORITHM')


def simpledecode(str):
    base = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
    subbed = 'QWERTYUIOPASDFGHJKLZXCVBNM'
    subdict = {}
    for i in xrange(0, len(base)):
        subdict[base[i]] = subbed[i]

    result = ''
    for c in str:
        if c in subdict:
            result += subdict[c]
        else:
            result += '?'
    return result


def main():
    unittest.main()


if __name__ == '__main__':
    main()