Move emotions to Face class.

Author: daspecster

packages/google-cloud-vision/google/cloud/vision/face.py

@@ -101,44 +101,44 @@ def from_api_repr(cls, response):
                    anger_likelihood)
 
     @property
-    def joy_likelihood(self):
-        """Likelihood of joy in detected face.
+    def anger(self):
+        """Likelihood of anger in detected face.
 
         :rtype: str
         :returns: String derived from
                   :class:`~google.cloud.vision.face.Likelihood`.
         """
-        return self._joy_likelihood
+        return self._anger_likelihood
 
     @property
-    def sorrow_likelihood(self):
-        """Likelihood of sorrow in detected face.
+    def joy(self):
+        """Likelihood of joy in detected face.
 
         :rtype: str
         :returns: String derived from
                   :class:`~google.cloud.vision.face.Likelihood`.
         """
-        return self._sorrow_likelihood
+        return self._joy_likelihood
 
     @property
-    def surprise_likelihood(self):
-        """Likelihood of surprise in detected face.
+    def sorrow(self):
+        """Likelihood of sorrow in detected face.
 
         :rtype: str
         :returns: String derived from
                   :class:`~google.cloud.vision.face.Likelihood`.
         """
-        return self._surprise_likelihood
+        return self._sorrow_likelihood
 
     @property
-    def anger_likelihood(self):
-        """Likelihood of anger in detected face.
+    def surprise(self):
+        """Likelihood of surprise in detected face.
 
         :rtype: str
         :returns: String derived from
                   :class:`~google.cloud.vision.face.Likelihood`.
         """
-        return self._anger_likelihood
+        return self._surprise_likelihood
 
 
 class Face(object):
@@ -182,6 +182,16 @@ def from_api_repr(cls, response):
                    headwear_likelihood, image_properties, landmarks,
                    landmarking_confidence)
 
+    @property
+    def anger(self):
+        """Accessor to likelihood that the detected face is angry.
+
+        :rtype: str
+        :returns: String derived from
+                  :class:`~google.cloud.vision.face.Likelihood`.
+        """
+        return self.emotions.anger
+
     @property
     def angles(self):
         """Accessor to the pan, tilt and roll angles of a Face.
@@ -230,7 +240,7 @@ def fd_bounds(self):
         return self._fd_bounds
 
     @property
-    def headwear_likelihood(self):
+    def headwear(self):
         """Headwear likelihood.
 
         :rtype: :class:`~google.cloud.vision.face.Likelihood`
@@ -248,6 +258,16 @@ def image_properties(self):
         """
         return self._image_properties
 
+    @property
+    def joy(self):
+        """Likelihood of joy in detected face.
+
+        :rtype: str
+        :returns: String derived from
+                  :class:`~google.cloud.vision.face.Likelihood`.
+        """
+        return self.emotions.joy
+
     @property
     def landmarks(self):
         """Accessor to the facial landmarks detected in a face.
@@ -267,6 +287,26 @@ def landmarking_confidence(self):
         """
         return self._landmarking_confidence
 
+    @property
+    def sorrow(self):
+        """Likelihood of sorrow in detected face.
+
+        :rtype: str
+        :returns: String derived from
+                  :class:`~google.cloud.vision.face.Likelihood`.
+        """
+        return self.emotions.sorrow
+
+    @property
+    def surprise(self):
+        """Likelihood of surprise in detected face.
+
+        :rtype: str
+        :returns: String derived from
+                  :class:`~google.cloud.vision.face.Likelihood`.
+        """
+        return self.emotions.surprise
+
 
 class FaceImageProperties(object):
     """A representation of the image properties from face detection."""
@@ -289,7 +329,7 @@ def from_api_repr(cls, response):
         return cls(blurred_likelihood, underexposed_likelihood)
 
     @property
-    def blurred_likelihood(self):
+    def blurred(self):
         """Likelihood of the image being blurred.
 
         :rtype: str
@@ -299,7 +339,7 @@ def blurred_likelihood(self):
         return self._blurred_likelihood
 
     @property
-    def underexposed_likelihood(self):
+    def underexposed(self):
         """Likelihood that the image used for detection was underexposed.
 
         :rtype: str

packages/google-cloud-vision/unit_tests/test_face.py

@@ -44,13 +44,13 @@ def test_face_landmarks(self):
     def test_facial_emotions(self):
         from google.cloud.vision.face import Likelihood
         self.assertEqual(Likelihood.VERY_LIKELY,
-                         self.face.emotions.joy_likelihood)
+                         self.face.joy)
         self.assertEqual(Likelihood.VERY_UNLIKELY,
-                         self.face.emotions.sorrow_likelihood)
+                         self.face.sorrow)
         self.assertEqual(Likelihood.VERY_UNLIKELY,
-                         self.face.emotions.surprise_likelihood)
+                         self.face.surprise)
         self.assertEqual(Likelihood.VERY_UNLIKELY,
-                         self.face.emotions.anger_likelihood)
+                         self.face.anger)
 
     def test_faciale_angles(self):
         self.assertEqual(-0.43419784, self.face.angles.roll)
@@ -60,11 +60,11 @@ def test_faciale_angles(self):
     def test_face_headware_and_blur_and_underexposed(self):
         from google.cloud.vision.face import Likelihood
         self.assertEqual(Likelihood.VERY_UNLIKELY,
-                         self.face.image_properties.blurred_likelihood)
+                         self.face.image_properties.blurred)
         self.assertEqual(Likelihood.VERY_UNLIKELY,
-                         self.face.headwear_likelihood)
+                         self.face.headwear)
         self.assertEqual(Likelihood.VERY_UNLIKELY,
-                         self.face.image_properties.underexposed_likelihood)
+                         self.face.image_properties.underexposed)
 
     def test_face_bounds(self):
         self.assertEqual(4, len(self.face.bounds.vertices))