def test_image_properties_annotation_from_pb(self):
from google.cloud.proto.vision.v1 import image_annotator_pb2
from google.protobuf.wrappers_pb2 import FloatValue
from google.type.color_pb2 import Color
alpha = FloatValue(value=1.0)
color_pb = Color(red=1.0, green=2.0, blue=3.0, alpha=alpha)
color_info_pb = image_annotator_pb2.ColorInfo(color=color_pb,
score=1.0,
pixel_fraction=1.0)
dominant_colors = image_annotator_pb2.DominantColorsAnnotation(
colors=[color_info_pb])
image_properties_pb = image_annotator_pb2.ImageProperties(
dominant_colors=dominant_colors)
color_info = self._get_target_class()
image_properties = color_info.from_pb(image_properties_pb)
self.assertEqual(image_properties.colors[0].pixel_fraction, 1.0)
self.assertEqual(image_properties.colors[0].score, 1.0)
self.assertEqual(image_properties.colors[0].color.red, 1.0)
self.assertEqual(image_properties.colors[0].color.green, 2.0)
self.assertEqual(image_properties.colors[0].color.blue, 3.0)
self.assertEqual(image_properties.colors[0].color.alpha, 1.0)
def test_empty_pb_rgb_color_data(self):
from google.type.color_pb2 import Color
color_pb = Color()
color_class = self._get_target_class()
color = color_class.from_pb(color_pb)
self.assertEqual(color.red, 0.0)
self.assertEqual(color.green, 0.0)
self.assertEqual(color.blue, 0.0)
self.assertEqual(color.alpha, 0.0)
def test_pb_rgb_color_no_alpha_data(self):
from google.protobuf.wrappers_pb2 import FloatValue
from google.type.color_pb2 import Color
alpha = FloatValue()
color_pb = Color(red=1.0, green=2.0, blue=3.0, alpha=alpha)
color_class = self._get_target_class()
color = color_class.from_pb(color_pb)
self.assertEqual(color.red, 1.0)
self.assertEqual(color.green, 2.0)
self.assertEqual(color.blue, 3.0)
self.assertEqual(color.alpha, 0.0)
