def test_Simulator_with_HandGenerator(self):
"""Test Simulator with HandGenerator."""
hands = Hands()
hands.addHand(HoldEm.Hand.fromString("AC 2C"))
hands.addHand(HoldEm.Hand.fromString("AH KH"))
hands.addHand(HoldEm.HandGenerator(SlanskyHand['class1']))
hands.addHand(HoldEm.HandGenerator(SlanskyHand['class2']))
hands.addHand(HoldEm.HandGenerator(SlanskyHand['class3']))
hands.addHand(HoldEm.HandGenerator(SlanskyHand['class4']))
simulator = HoldEm.Simulator(predefined_hands=hands)
simulator.simulate_games(number_of_games=10)
def main(argv=None):
if argv is None:
argv = sys.argv
usage = "usage: %prog [<options>]"
version = "%prog version 1.0"
parser = OptionParser(usage)
parser.add_option("-B", "--board", type="string", dest="board",
metavar="cards", help="specify the flop")
parser.add_option("-g", "--game", type="string", dest="game",
default="holdem", help="game to simulate")
parser.add_option("-H", "--hand", type="string", dest="hands",
metavar="cards", action="append", help="add a hand")
parser.add_option("-n", "--numGames", type="int", dest="numGames",
default=100, help="number of games to simulate")
parser.add_option("-N", "--numHands", type="int", dest="numHands",
default=10, help="number of hands in play")
parser.add_option("-p", "--showProgress", action="store_true",
dest="showProgress", default=False, help="show progress")
parser.add_option("-P", "--profile", type="string", dest="profile",
default=None, help="enable profiling")
parser.add_option("-q", "--quiet", action="store_true",
dest="quiet", default=False, help="run quietly")
parser.add_option("-v", "--verbose", action="store_true", dest="verbose",
default=False, help="show results of each hand")
(options, args) = parser.parse_args()
game = {
"holdem" : HoldEm.Game,
"5cardstud" : FiveCardStud.Game,
"fivecardstud" : FiveCardStud.Game,
"5cardstudhilo" : FiveCardStud.HiLoGame,
"fivecardstudhilo" : FiveCardStud.HiLoGame,
"7cardstud" : SevenCardStud.Game,
"sevencardstud" : SevenCardStud.Game,
"7cardstudhilo" : SevenCardStud.HiLoGame,
"sevencardstudhilo" : SevenCardStud.HiLoGame,
"omaha" : Omaha.Game,
"omahahilo" : Omaha.HiLoGame,
}
if game.has_key(options.game):
GameClass = game[options.game]
else:
print "Unknown game type \"%s\"" % options.game
print "Known games are:"
for name in game.keys():
print "\t%s" % name
sys.exit(1)
maxHands = GameClass.getMaxHands()
if options.numHands > maxHands:
options.numHands = maxHands
if options.verbose:
print "Reducing number of hands to %d" % maxHands
game = GameClass(numHands = options.numHands)
HandClass = GameClass.getHandClass()
if options.hands is not None:
hands = Hands()
for hand in options.hands:
hands.addHand(HandClass.fromString(hand))
game.addHands(hands)
if options.board:
game.setBoard(Board.fromString(options.board))
if options.verbose:
callback=showHandCallback
print "Simulating %d games of %s" % (options.numGames,
GameClass.gameName)
print "%d Hands" % game.getNumHands(),
if game.hands:
print ": %s" % game.hands
else:
print
if game.board:
print "Board: %s" % game.board
elif options.showProgress:
callback=showProgressCallback
else:
callback=None
cmd="game.simulateGames(options.numGames, callback=callback)"
if options.profile:
import cProfile
if options.verbose:
print "Profiling to file %s" % options.profile
# Need to supply context here as run() will just use __main__
cProfile.runctx(cmd, globals(), locals(), filename=options.profile)
else:
eval(cmd)
if options.showProgress:
print
if not options.quiet:
print game.statsToString()
