def setUp(self):
self.model = ChainModel()
self.act_a = self.model.act_a
self.act_b = self.model.act_b
self.s1 = self.model.get_state_by_id(1)
self.s2 = self.model.get_state_by_id(2)
self.s3 = self.model.get_state_by_id(3)
def parse_model(s):
if s == 'cm':
return ChainModel()
elif s == 'scm':
return SlipperyChainModel()
elif s == 'lm':
return LoopModel()
else:
invalid()
def get_model(model_name):
if model_name == "Chain":
return ChainModel()
elif model_name == "SlipperyChain":
return SlipperyChainModel()
elif model_name == "Chain2":
return ChainModel2()
elif model_name == "Loop":
return LoopModel()
elif model_name == "LoopDeadEnd":
return LoopModelDeadEnd()
elif model_name == "LoopDiffTrans":
return LoopModelDiffTrans()
elif model_name == "SpecialLoop":
return SpecialLoopModel()
else:
raise Exception(model_name+ " not found")
