Merge pull request #107 from rosette-api/RCB-601-events-endpoint

RCB-601 new: events endpoint

Author: davidliu-github

rosette/api.py

@@ -589,7 +589,8 @@ def __init__(
             'TEXT_EMBEDDING': 'semantics/vector',
             'TOKENS': 'tokens',
             'TOPICS': 'topics',
-            'TRANSLITERATION': 'transliteration'
+            'TRANSLITERATION': 'transliteration',
+            'EVENTS': 'events'
         }
 
     def __del__(self):
@@ -1010,3 +1011,13 @@ def similar_terms(self, parameters):
         :return: A python dictionary containing the similar terms and their similarity
         """
         return EndpointCaller(self, self.endpoints['SIMILAR_TERMS']).call(parameters)
+
+    def events(self, parameters):
+        """
+        Create an L{EndpointCaller} to identify events found in the texts.
+        @param parameters: An object specifying the data,
+        and possible metadata, to be processed by the 'events' identifier.
+        @type parameters: L{DocumentParameters} or L{str}
+        @return: A python dictionary containing the results of event extraction.
+        """
+        return EndpointCaller(self, self.endpoints['EVENTS']).call(parameters)

tests/test_rosette_api.py

@@ -882,3 +882,19 @@ def test_the_deprecated_endpoints(api, json_response, doc_params):
 
     httpretty.disable()
     httpretty.reset()
+
+# Test the events endpoint
+
+
+def test_the_events_endpoint(api, json_response, doc_params):
+    """Test the events endpoint"""
+    httpretty.enable()
+    httpretty.register_uri(httpretty.POST, "https://api.rosette.com/rest/v1/info",
+                           body=json_response, status=200, content_type="application/json")
+    httpretty.register_uri(httpretty.POST, "https://api.rosette.com/rest/v1/events",
+                           body=json_response, status=200, content_type="application/json")
+
+    result = api.events(doc_params)
+    assert result["name"] == "Rosette"
+    httpretty.disable()
+    httpretty.reset()