diff --git a/src/tests/testGo.py b/src/tests/testGo.py --- a/src/tests/testGo.py +++ b/src/tests/testGo.py @@ -1,6 +1,6 @@ from unittest import TestCase -from go.core import isLegalPosition +from go.core import isLegalPosition, Go class TestLegal(TestCase): @@ -23,3 +23,25 @@ class TestLegal(TestCase): [0,0,0,0,0] ] self.assertFalse(isLegalPosition(board)) + + +class TestMove(TestCase): + def testCapture(self): + g=Go(3) + g.load([ + [0,1,0], + [1,-1,0], + [0,1,0] + ]) + g.toMove=1 + g.doMove(1,1,2) + self.assertEqual(g.board,[ + [0,1,0], + [1,0,1], + [0,1,0] + ]) + + g._helper.clear() + for row in g._helper._board: + for x in row: + self.assertEqual(x,0)