def create_characterization_model_dictionary(self):
characterization_dictionary = {}
self.model_catalog.add_models_from_model_catalog_db(self.model_catalog_db_data_io)
# TODO - only show valid models - FYI model_catalog.add_models already does a check for valid model(DCA)
characterization_models = self.model_catalog.characterization_models()
for model in characterization_models:
characterization_string = model.model_path + " " + Utility.format_date(model.create_date) + " " + model.created_by + " " + str(model.id)
characterization_dictionary[characterization_string] = model
self.characterization_model = characterization_dictionary
def test_format_date_string_calls_strftime_with_correct_date_string_argument(self):
date = mock.MagicMock(datetime.date)
date.strftime.return_value = "date string"
return_string = Utility.format_date(date)
self.assertEquals(return_string, "date string")
self.assertEquals("%m/%d/%Y %H:%M %p", date.strftime.call_args[0][0])
def create_alternative_model_dictionary(self):
alternative_dictionary = {}
self.model_catalog.add_models_from_model_catalog_db(
self.model_catalog_db_data_io)
alternative_models = self.model_catalog.alternative_models()
for model in alternative_models:
alternative_string = model.model_name + " " + model.model_path + " " + model.created_by + " " + Utility.format_date(
model.create_date)
alternative_dictionary[alternative_string] = model
self.alternative_model_dict = alternative_dictionary
def create_characterization_model_dictionary(self):
characterization_dictionary = {}
self.model_catalog.add_models_from_model_catalog_db(
self.model_catalog_db_data_io)
characterization_models = self.model_catalog.characterization_models()
for model in characterization_models:
characterization_string = model.model_name + " " + model.model_path + " " + model.created_by + " " + Utility.format_date(
model.create_date)
characterization_dictionary[characterization_string] = model
self.characterization_model_dict = characterization_dictionary
def create_non_calibration_model_dictionary(self):
non_calibration_dictionary = {}
self.model_catalog.add_models_from_model_catalog_db(
self.model_catalog_db_data_io)
non_calibration_models = self.model_catalog.non_calibration_models()
for model in non_calibration_models:
non_calibration_string = model.model_name + " " + model.model_path + " " + model.created_by + " " + Utility.format_date(
model.create_date)
non_calibration_dictionary[non_calibration_string] = model
self.non_calibration_model_dict = non_calibration_dictionary
def create_registered_model_dictionary(self):
model_dictionary = {}
self.model_catalog.add_models_from_model_catalog_db(
self.model_catalog_db_data_io)
for model in self.model_catalog.models:
model_string = model.model_name + " " + model.model_path + " " + model.created_by + " " + Utility.format_date(
model.create_date)
model_dictionary[model_string] = model
self.registered_model_dict = model_dictionary