def test_SetupDeterministicTransitionByStateSet2Agent( self, jointState, jointAction, expectedResult): getTransitionTable = targetCode.SetupDeterministicTransitionByStateSet2Agent( self.stateSet4x4, self.cardinalActionSet, self.goalState) transitionTable = getTransitionTable() nextStateDistribution = transitionTable[jointState][jointAction] self.assertEqual(nextStateDistribution, expectedResult)
def setUp(self): cardinalActionSet = [(-1,0), (0,1), (1,0), (0,-1), (0,0)] gridWidth = 4 gridHeight = 4 self.goalState = (3,3) self.trapState = (0,0) stateSet4x4 = list(itertools.product(range(gridWidth), range(gridHeight))) getTransitionTable = targetCode.SetupDeterministicTransitionByStateSet2Agent(stateSet4x4,cardinalActionSet, self.goalState) self.transitionTable = getTransitionTable()