Submission #213474

---

Source Code Expand

#!/usr/bin/env python2.7

import sys
from collections import namedtuple

from cStringIO import StringIO
import unittest
import cProfile

def main():
    X, = (str(x) for x in sys.stdin.readline().split())
    print solve(X)

def solve(X):
    d = {'A': 1, 'B': 2, 'C':3, 'D':4, 'E':5}
    return d[X]
    
class Test(unittest.TestCase):

    @staticmethod
    def tryone(indata):
        sys.stdin = StringIO(indata)
        out = sys.stdout = StringIO()
        main()
        return out.getvalue()

    def test50(self):
        self.assertEqual(solve('A'), 1)
        self.assertEqual(solve('B'), 2)
        self.assertEqual(solve('C'), 3)
        self.assertEqual(solve('D'), 4)
        self.assertEqual(solve('E'), 5)

    def test90(self):
        self.assertEqual(self.tryone("""\
A
"""), """1\n""")

if __name__ == '__main__':
    if len(sys.argv) > 1:
        print "_/" * 30 + str(sys.argv)
        if sys.argv[1] == '-p':
            sys.argv.pop(1)
            cProfile.run("unittest.main(exit=False, failfast=True)", sort='time')
        else:
            unittest.main()
    else:
        main()

Submission Info

Judge Result