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