diff --git a/games.py b/games.py index 6aded01d5..ae922f0e5 100644 --- a/games.py +++ b/games.py @@ -417,7 +417,13 @@ def __init__(self, h=7, v=6, k=4): def actions(self, state): return [(x, y) for (x, y) in state.moves - if y == 1 or (x, y - 1) in state.board] + if x == self.h or (x + 1 , y ) in state.board] + +class Gomoku(TicTacToe): + """Also known as Five in a row.""" + + def __init__(self, h=15, v=16, k=5): + TicTacToe.__init__(self, h, v, k) class Backgammon(StochasticGame):