class InitStateTestCase(unittest.TestCase): def setUp(self): self.initboard = choice(test_boards) self.state = State(self.initboard) def testload(self): ''' compares input board with printed representation of state ''' self.assertEqual(self.state.__str__().strip(), self.initboard.strip()) def testgenmoves(self): for move in gen_moves(self.state): self.state.move(move) self.state.unmove() InitStateTestCase.testload(self)
class SearchTestCase(unittest.TestCase): def setUp(self): self.initboard = test_boards[3] #self.initboard = choice(test_boards) self.state = State(self.initboard) def testnegamax(self): ''' test deep negamax search ''' m, d, t = negamax(self.state, t=3.0) print ' d = %d, time = %ds' % (d, t) self.assertEqual(self.state.__str__().strip(), self.initboard.strip()) def testalphabeta(self): m, d, t = alphabeta(self.state, t=3.0) print ' d = %d time = %ds' % (d, t) self.assertEqual(self.state.__str__().strip(), self.initboard.strip())
def setUp(self): self.initboard = test_boards[3] #self.initboard = choice(test_boards) self.state = State(self.initboard)