def test_brain_update(mock_get):
""" Tests brain update function """
# Patch and Mock out requests/responses
response = Mock()
test_json = {'versions': [{'version': 1}], 'state': 'DONE'}
response.json.return_value = test_json
mock_get.return_value = response
config = Config([__name__])
brain = Brain(config)
assert mock_get.call_count == 2
brain.update()
assert mock_get.call_count == 4
def _run_bonsai(args):
logging.debug("_run_bonsai starting")
config = Config(args)
logging.debug(config)
brain = Brain(config)
brain.update()
sim = SimulinkSimulation(brain, "simulink_sim")
logging.info('%s running' % brain.name)
while sim.run():
continue
logging.info('%s finished' % brain.name)
def test_brain_predict_exist_ready(predict_config, v2_get):
""" Tests existence and readiness in predict mode """
brain = Brain(predict_config)
brain.update()
assert brain.exists is True
assert brain.ready is True
if __name__ == "__main__":
# parse configuration options specific to this simulation
# additional args will be read with Config(sys.argv)
parser = argparse.ArgumentParser(description='Launch EnergyPlus simulator')
parser.add_argument('--test_croom', action='store_true', default=False)
parser.add_argument('--test_energyplus',
action='store_true',
default=False)
args, unknown = parser.parse_known_args()
# Test the results from the original models...
if args.test_croom or args.test_energyplus:
if args.test_croom:
test_model(model=CRoom())
elif args.test_energyplus:
test_model(model=ePlus85Actuator())
# ...or run the AI for simulation
else:
config = Config(sys.argv)
brain = Brain(config)
sim = EnergyPlusSimulator(brain, 'energyplus_simulator')
brain.update()
while (sim.run()):
continue
# finished, dump out plot
sim.finished()