def test_get_entity_type_from_class_name(self): from trustedanalytics.meta.names import class_name_to_entity_type self.assertEqual("model:logistic_regression", class_name_to_entity_type("LogisticRegressionModel")) self.assertEqual("model", class_name_to_entity_type("_BaseModel")) with self.assertRaises(ValueError) as cm: class_name_to_entity_type("") self.assertEqual(str(cm.exception), "Invalid empty class_name, expected non-empty string")
def _generate_full_name(class_name, member_name): entity_type = class_name_to_entity_type(class_name) full_name = "%s/%s" % (entity_type, member_name) return full_name