def setUp(self):
"""Set up decay model"""
#Read the full SM
sm_path = import_ufo.find_ufo_path('sm')
self.base_model = import_ufo.import_full_model(sm_path)
model = copy.deepcopy(self.base_model)
self.model = import_ufo.RestrictModel(model)
self.restrict_file = os.path.join(_file_path, os.path.pardir,
'input_files', 'restrict_sm.dat')
def setUp(self):
""" prepare a model and the ParamCardWriter"""
# load the SM with restriction
self.model = import_ufo.import_model('sm-full')
self.model = import_ufo.RestrictModel(self.model)
self.restrict_file = os.path.join(_file_path, os.path.pardir,
'input_files', 'restrict_sm.dat')
self.model.restrict_model(self.restrict_file)
# initialize the main object
self.writter = writter.ParamCardWriter(self.model)
self.content = StringIO.StringIO()
self.writter.define_output_file(self.content)
self.content.truncate(0) # remove the header