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)