chore(style): run yapf on tests

Author: ehdsouza

test/unit/test_assistant_v1.py

@@ -28,6 +28,7 @@
 ##############################################################################
 # region
 
+
 #-----------------------------------------------------------------------------
 # Test Class for create_session
 #-----------------------------------------------------------------------------
@@ -73,16 +74,16 @@ def make_url(self, body):
 
     def add_mock_response(self, url, response):
         responses.add(responses.POST,
-                    url,
-                    body=json.dumps(response),
-                    status=201,
-                    content_type='application/json')
-    
+                      url,
+                      body=json.dumps(response),
+                      status=201,
+                      content_type='application/json')
+
     def call_service(self, body):
         service = AssistantV2(
             authenticator=NoAuthAuthenticator(),
             version='2019-02-28',
-            )
+        )
         service.set_service_url(base_url)
         output = service.create_session(**body)
         return output
@@ -137,22 +138,23 @@ def test_delete_session_empty(self):
     #- Helpers -
     #-----------
     def make_url(self, body):
-        endpoint = '/v2/assistants/{0}/sessions/{1}'.format(body['assistant_id'], body['session_id'])
+        endpoint = '/v2/assistants/{0}/sessions/{1}'.format(
+            body['assistant_id'], body['session_id'])
         url = '{0}{1}'.format(base_url, endpoint)
         return url
 
     def add_mock_response(self, url, response):
         responses.add(responses.DELETE,
-                    url,
-                    body=json.dumps(response),
-                    status=200,
-                    content_type='')
-    
+                      url,
+                      body=json.dumps(response),
+                      status=200,
+                      content_type='')
+
     def call_service(self, body):
         service = AssistantV2(
             authenticator=NoAuthAuthenticator(),
             version='2019-02-28',
-            )
+        )
         service.set_service_url(base_url)
         output = service.delete_session(**body)
         return output
@@ -180,6 +182,7 @@ def construct_required_body(self):
 ##############################################################################
 # region
 
+
 #-----------------------------------------------------------------------------
 # Test Class for message
 #-----------------------------------------------------------------------------
@@ -219,22 +222,23 @@ def test_message_empty(self):
     #- Helpers -
     #-----------
     def make_url(self, body):
-        endpoint = '/v2/assistants/{0}/sessions/{1}/message'.format(body['assistant_id'], body['session_id'])
+        endpoint = '/v2/assistants/{0}/sessions/{1}/message'.format(
+            body['assistant_id'], body['session_id'])
         url = '{0}{1}'.format(base_url, endpoint)
         return url
 
     def add_mock_response(self, url, response):
         responses.add(responses.POST,
-                    url,
-                    body=json.dumps(response),
-                    status=200,
-                    content_type='application/json')
-    
+                      url,
+                      body=json.dumps(response),
+                      status=200,
+                      content_type='application/json')
+
     def call_service(self, body):
         service = AssistantV2(
             authenticator=NoAuthAuthenticator(),
             version='2019-02-28',
-            )
+        )
         service.set_service_url(base_url)
         output = service.message(**body)
         return output
@@ -243,7 +247,18 @@ def construct_full_body(self):
         body = dict()
         body['assistant_id'] = "string1"
         body['session_id'] = "string1"
-        body.update({"input": MessageInput._from_dict(json.loads("""{"message_type": "fake_message_type", "text": "fake_text", "options": {"debug": false, "restart": false, "alternate_intents": false, "return_context": true}, "intents": [], "entities": [], "suggestion_id": "fake_suggestion_id"}""")), "context": MessageContext._from_dict(json.loads("""{"global": {"system": {"timezone": "fake_timezone", "user_id": "fake_user_id", "turn_count": 10}}, "skills": {}}""")), })
+        body.update({
+            "input":
+                MessageInput._from_dict(
+                    json.loads(
+                        """{"message_type": "fake_message_type", "text": "fake_text", "options": {"debug": false, "restart": false, "alternate_intents": false, "return_context": true}, "intents": [], "entities": [], "suggestion_id": "fake_suggestion_id"}"""
+                    )),
+            "context":
+                MessageContext._from_dict(
+                    json.loads(
+                        """{"global": {"system": {"timezone": "fake_timezone", "user_id": "fake_user_id", "turn_count": 10}}, "skills": {}}"""
+                    )),
+        })
         return body
 
     def construct_required_body(self):
@@ -275,6 +290,7 @@ def check_empty_required_params(obj, response):
         error = True
     assert error
 
+
 def check_missing_required_params(obj):
     """Test function to assert that the operation will throw an error when missing required data
 
@@ -291,6 +307,7 @@ def check_missing_required_params(obj):
         error = True
     assert error
 
+
 def check_empty_response(obj):
     """Test function to assert that the operation will return an empty response when given an empty request
 
@@ -302,6 +319,7 @@ def check_empty_response(obj):
     url = obj.make_url(body)
     send_request(obj, {}, {}, url=url)
 
+
 def send_request(obj, body, response, url=None):
     """Test function to create a request, send it, and assert its accuracy to the mock response
 
@@ -318,6 +336,7 @@ def send_request(obj, body, response, url=None):
     assert responses.calls[0].request.url.startswith(url)
     assert output.get_result() == response
 
+
 ####################
 ## Mock Responses ##
 ####################

test/unit/test_assistant_v2.py

@@ -17,11 +17,16 @@
 from ibm_watson import get_sdk_headers
 import unittest
 
+
 class TestCommon(unittest.TestCase):
+
     def test_get_sdk_headers(self):
         headers = get_sdk_headers('my_service', 'v1', 'my_operation')
         self.assertIsNotNone(headers)
         self.assertIsNotNone(headers.get('X-IBMCloud-SDK-Analytics'))
         self.assertIsNotNone(headers.get('User-Agent'))
         self.assertIn('watson-apis-python-sdk', headers.get('User-Agent'))
-        self.assertEqual(headers.get('X-IBMCloud-SDK-Analytics'), 'service_name=my_service;service_version=v1;operation_id=my_operation')
+        self.assertEqual(
+            headers.get('X-IBMCloud-SDK-Analytics'),
+            'service_name=my_service;service_version=v1;operation_id=my_operation'
+        )