def test_map_resource_to_metadata_type_with_serializable_type(self): mb_sdk_type = List parameter_name, parameter_type = utils.map_resource_to_metadata_type( mb_sdk_type) self.assertEqual(parameter_name, 'exported_dataset') self.assertEqual(parameter_type, 'JsonArray')
def test_map_resource_to_metadata_type_with__Dataset_type(self): mb_sdk_type = '_Dataset' parameter_name, parameter_type = utils.map_resource_to_metadata_type( mb_sdk_type) self.assertEqual(parameter_name, 'dataset') self.assertEqual(parameter_type, 'Dataset')
def test_map_resource_to_metadata_type_with_mb_sdk_type(self): mb_sdk_type = aiplatform.Model parameter_name, parameter_type = utils.map_resource_to_metadata_type( mb_sdk_type) self.assertEqual(parameter_name, 'model') self.assertEqual(parameter_type, 'Model')