def test_whenRunSolveMazeIsCalledNoTrialsTimes(self):
noTrials = 2
noIterations = 2
performance = {iteration: PerformanceRecord([0]) for iteration in range(noIterations)}
mockedAgent = Agent(None, None, None)
mockedAgent.solveMaze = MagicMock(return_value=performance)
mockedAgentFactory = AgentFactory(None,None,None)
mockedAgentFactory.createAgent = MagicMock(return_value=mockedAgent)
experiment = Experiment(mockedAgentFactory)
experiment.run(1, noTrials)
mockedAgent.solveMaze.assert_called(noTrials)
import traceback
if __name__ == "__main__":
try:
if len(sys.argv) != 6:
raise Exception("Wrong number of arguments, see usage")
expType = sys.argv[1]
problemName = sys.argv[2]
modelName = sys.argv[3]
noTrials = int(sys.argv[4])
noIterations = int(sys.argv[5])
experimentName = "{}-{}-{}.json".format(expType, modelName, problemName)
agentFactory = AgentFactory(expType, problemName, modelName)
experiment = Experiment(agentFactory)
experiment.run(noIterations, noTrials)
experiment.saveResults(experimentName)
print "Results saved in file {}".format(experimentName)
except:
print traceback.format_exc()
print "Usage: runExperiment {}|{} {}|{}|{} {}|{} noTrials noIterations".format(AgentFactory.PRED_TYPE,
AgentFactory.BEH_TYPE,
AgentFactory.BASIC_PBM,
AgentFactory.SEC_ORDER_PBM,
AgentFactory.MULTI_STEP_PBM,
AgentFactory.HTM_MODEL,
AgentFactory.Q_MODEL)
