Python: 渐进猜数字游戏 <8> 代码优化 与 结项

'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
# Copyright (c) 2016,Liexun Studio
# All Rights Reserved.
#
# File Name: GunessNum.py
# Summary: Guness Num Game
#   1. Generate random number and check the number correction
#   2. Get user input and check the number correction
#   3. Compare random and user input, output result
#
# Version: 1.0
# Author: Roy
# End Date: 08/15/2016
#
# Update Version: 1.X
# Author:
# End Date:
'''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''

#encode="utf-8"

import random

GC = 5      #define try account for guess
randNum = 0

def startGame():
print "*************************************************************"
print "************************ Guess Number ***********************"
print "*************************************************************"

def randNumber():
#Get system random number: 4 bits, without duplication
randNum = random.sample(xrange(0,9),4)
##print type(randNum)
print "\nSystem random number: " + str(randNum) + "\n"
return randNum

#check the correction of number
def checkNum(inNum):
#Is number?
if inNum.isdigit() == True:
#4 digit?
if len(inNum) == 4:
#duplicate?
for i in range(4):
if inNum.count(inNum[i]) == 1:
if i != 3:
pass
else:
print "Correct!"
return True
else:
print "Duplicate, re-input."
return False
break
else:
print "Not 4 digit, re-input."
return False
else:
print "Not number, re-input."
return False

#compare and output result: XAXB
def compNum(rnum,inum):
ca = 0
cb = 0
for i in range(4):
if inum[i] in rnum:
if rnum[i] == inum[i]:
ca += 1
else:
cb += 1
return str(ca) + "A" + str(cb) + "B"

#try to guess number as system define: GC
def gunessNumber():
randNum = randNumber()
inNum = raw_input(u"Input your guess number: ")   #get user input
for gc in range(GC):
while not checkNum(inNum):                    #get user input and check the correction
inNum = raw_input(u"Input your guess number: ")
else:
print str(gc+1) + "--" + compNum(map(lambda x: str(x), randNum),inNum)   #ouput the result
if compNum(map(lambda x: str(x), randNum),inNum) == "4A0B":              #iteration until guess the correct numer or achieve the system define number
print "Congratulations!!!"
break
elif gc < 4:
inNum = raw_input(u"Input your guess number: ")
else:
print "Unfortunately, Faild, Please try again."

if __name__ == '__main__':
startGame()
gunessNumber()

#encode="utf-8"

import unittest
from GunessNum import checkNum

class Test_GunessNumber(unittest.TestCase):
#Executed before each test, initialize test
def setup():
pass

#Executed after each test, clear up test
def teardown():
pass

def test_checknumber(self):
self.assertFalse(checkNum("123"))
self.assertFalse(checkNum("1233"))
self.assertFalse(checkNum("12356"))
self.assertTrue(checkNum("1234"))

if __name__ == '__main__':
test_suite = unittest.TestSuite()
test_suite.addTest(Test_GunessNumber("test_checknumber"))

runner = unittest.TextTestRunner()
runner.run(test_suite)

Python 2.7.11 (v2.7.11:6d1b6a68f775, Dec  5 2015, 20:40:30) [MSC v.1500 64 bit (AMD64)] on win32
Type "copyright", "credits" or "license()" for more information.
>>>
=================== RESTART: G:\Project\Python\Test_GN.py ===================
Not 4 digit, re-input.
Duplicate, re-input.
Not 4 digit, re-input.
Correct!
.
----------------------------------------------------------------------
Ran 1 test in 0.020s

OK
>>>