Adding Sentiment class for language.

dhermes · dhermes · commit c70c4710ff93 · 2016-08-24T15:03:34.000-07:00
diff --git a/docs/language-responses.rst b/docs/language-responses.rst
@@ -7,3 +7,10 @@ Entity
 .. automodule:: gcloud.language.entity
   :members:
   :show-inheritance:
+
+Sentiment
+~~~~~~~~~
+
+.. automodule:: gcloud.language.sentiment
+  :members:
+  :show-inheritance:
diff --git a/gcloud/language/sentiment.py b/gcloud/language/sentiment.py
@@ -0,0 +1,41 @@
+# Copyright 2016 Google Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Definition for Google Cloud Natural Language API sentiment.
+
+Sentiment is the response to an ``analyzeSentiment`` request.
+"""
+
+
+class Sentiment(object):
+    """A Google Cloud Natural Language API sentiment object.
+
+    .. _Sentiment message: https://cloud.google.com/natural-language/\
+                           reference/rest/v1beta1/Sentiment
+
+    See `Sentiment message`_.
+
+    :type polarity: float
+    :param polarity: Polarity of the sentiment in the ``[-1.0, 1.0]`` range.
+                     Larger numbers represent more positive sentiments.
+
+    :type magnitude: float
+    :param magnitude: A non-negative number in the ``[0, +inf)`` range, which
+                      represents the absolute magnitude of sentiment
+                      regardless of polarity (positive or negative).
+    """
+
+    def __init__(self, polarity, magnitude):
+        self.polarity = polarity
+        self.magnitude = magnitude
diff --git a/gcloud/language/test_sentiment.py b/gcloud/language/test_sentiment.py
@@ -0,0 +1,32 @@
+# Copyright 2016 Google Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+
+
+class TestSentiment(unittest.TestCase):
+
+    def _getTargetClass(self):
+        from gcloud.language.sentiment import Sentiment
+        return Sentiment
+
+    def _makeOne(self, *args, **kw):
+        return self._getTargetClass()(*args, **kw)
+
+    def test_constructor(self):
+        polarity = 1
+        magnitude = 2.3
+        sentiment = self._makeOne(polarity, magnitude)
+        self.assertEqual(sentiment.polarity, polarity)
+        self.assertEqual(sentiment.magnitude, magnitude)
