def test_tone_with_positional_args():
tone_url = 'https://gateway.watsonplatform.net/tone-analyzer/api/v3/tone'
tone_args = {'version': '2016-05-19', 'sentences': 'false'}
tone_response = None
with open(
os.path.join(
os.path.dirname(__file__),
'../../resources/tone-v3-expect1.json')) as response_json:
tone_response = response_json.read()
responses.add(responses.POST,
tone_url,
body=tone_response,
status=200,
content_type='application/json')
with open(
os.path.join(os.path.dirname(__file__),
'../../resources/personality.txt')) as tone_text:
authenticator = BasicAuthenticator('username', 'password')
tone_analyzer = ibm_watson.ToneAnalyzerV3('2016-05-19',
authenticator=authenticator)
tone_analyzer.tone(tone_text.read(),
content_type='application/json',
sentences=False)
assert responses.calls[0].request.url.split('?')[0] == tone_url
# Compare args. Order is not deterministic!
actualArgs = {}
for arg in responses.calls[0].request.url.split('?')[1].split('&'):
actualArgs[arg.split('=')[0]] = arg.split('=')[1]
assert actualArgs == tone_args
assert responses.calls[0].response.text == tone_response
assert len(responses.calls) == 1
def test_error():
tone_url = 'https://gateway.watsonplatform.net/tone-analyzer/api/v3/tone'
error_code = 400
error_message = "Invalid JSON input at line 2, column 12"
tone_response = {
"code": error_code,
"sub_code": "C00012",
"error": error_message
}
responses.add(responses.POST,
tone_url,
body=json.dumps(tone_response),
status=error_code,
content_type='application/json')
authenticator = BasicAuthenticator('username', 'password')
tone_analyzer = ibm_watson.ToneAnalyzerV3('2016-05-19',
authenticator=authenticator)
text = 'Team, I know that times are tough!'
try:
tone_analyzer.tone(text, content_type='application/json')
except ApiException as ex:
assert len(responses.calls) == 1
assert isinstance(ex, ApiException)
assert ex.code == error_code
assert ex.message == error_message
def test_tone_chat():
tone_url = 'https://gateway.watsonplatform.net/tone-analyzer/api/v3/tone_chat'
tone_args = '?version=2016-05-19'
tone_response = None
with open(
os.path.join(
os.path.dirname(__file__),
'../../resources/tone-v3-expect2.json')) as response_json:
tone_response = response_json.read()
responses.add(responses.POST,
tone_url,
body=tone_response,
status=200,
content_type='application/json')
authenticator = BasicAuthenticator('username', 'password')
tone_analyzer = ibm_watson.ToneAnalyzerV3('2016-05-19',
authenticator=authenticator)
utterances = [{'text': 'I am very happy', 'user': '******'}]
tone_analyzer.tone_chat(utterances)
assert responses.calls[0].request.url == tone_url + tone_args
assert responses.calls[0].response.text == tone_response
assert len(responses.calls) == 1
def test_tone():
tone_url = 'https://gateway.watsonplatform.net/tone-analyzer/api/v3/tone'
tone_args = '?version=2016-05-19'
tone_response = None
with open(
os.path.join(
os.path.dirname(__file__),
'../../resources/tone-v3-expect1.json')) as response_json:
tone_response = response_json.read()
responses.add(responses.POST,
tone_url,
body=tone_response,
status=200,
content_type='application/json')
with open(
os.path.join(os.path.dirname(__file__),
'../../resources/personality.txt')) as tone_text:
tone_analyzer = ibm_watson.ToneAnalyzerV3("2016-05-19",
username="******",
password="******")
tone_analyzer.tone(tone_text.read(), content_type='application/json')
assert responses.calls[0].request.url == tone_url + tone_args
assert responses.calls[0].response.text == tone_response
assert len(responses.calls) == 1
########################################################################################################################
# Start Watson Assistant service using your credentials
########################################################################################################################
conversation_client = ibm_watson.AssistantV1(
iam_apikey = 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx', # replace with apikey
url = 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx', # replace with Watson Assistant's Credentials - URL
version = '2018-09-20'
)
workspace_id = 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx' # replace with Skill ID
########################################################################################################################
# Start Watson Tone Analyser service using your credentials
########################################################################################################################
tone_analyser = ibm_watson.ToneAnalyzerV3(
version='2017-09-21',
url='xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx', #replace with gateway URL
iam_apikey='xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx' #replace with apikey
)
########################################################################################################################
# Start Watson Discovery service using your credentials
########################################################################################################################
discovery_client = ibm_watson.DiscoveryV1(
version='2018-08-01',
url='https://gateway-syd.watsonplatform.net/discovery/api', #replace with gateway URL
iam_apikey='xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'#replace with apikey
)
########################################################################################################################
# Start Slack service using Bot Token
def setUp(self):
self.tone_analyzer = ibm_watson.ToneAnalyzerV3(version='2017-09-21')
self.tone_analyzer.set_default_headers({
'X-Watson-Learning-Opt-Out': '1',
'X-Watson-Test': '1'
})
