Submission #143459

---

Source Code Expand

#!/usr/bin/env python2.7

import sys
from collections import namedtuple

from cStringIO import StringIO
import unittest
import cProfile

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

def solve(S):
    S = S.replace('O', '0');
    S = S.replace('D', '0');
    S = S.replace('I', '1');
    S = S.replace('Z', '2');
    S = S.replace('S', '5');
    S = S.replace('B', '8');

    return S
    
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('1Z0'), '120')
        self.assertEqual(solve('4ZD6O'), '42060')
        self.assertEqual(solve('BI9Z'), '8192')

    def test90(self):
        self.assertEqual(self.tryone("""\
1Z0
"""), """120\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