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()