def test_analyze():
"""
Test Sentiment Analysis
:return:
"""
servicer = analysis.SentimentAnalysisServicer()
request = Message()
request.value = b64_sentences.senteces()
context = object()
response = servicer.Analyze(request, context)
decoded_result = base64.b64decode(response.value).decode('utf-8')
logger.debug(decoded_result)
assert "pos" in str(decoded_result) or "neg" in str(
decoded_result), "Generated result is not valid!"
def test_recognize():
"""
Test Named Entity Recognition
:return:
"""
servicer = ner.RecognizeMessageServicer()
request = Request()
request.value = b64_sentences.senteces()
context = object()
response = servicer.Recognize(request, context)
decoded_result = base64.b64decode(response.value).decode('utf-8')
if "PERSON" in str(decoded_result) or "ORGANIZATION" in str(decoded_result) or "LOCATION" in str(decoded_result):
logger.debug("test_recognize() - OK")
assert True
else:
assert False, "Generated result is not valid"
if __name__ == '__main__':
try:
logger.debug('call => __name == __main__')
# Service ONE - Sentiment Analysis
endpoint = 'localhost:{}'.format(registry['sentiment_analysis']['grpc'])
# Open a gRPC channel
channel = grpc.insecure_channel('{}'.format(endpoint))
except KeyError as e:
print(e)
try:
logger.debug("call => SentimentAnalysis() Service Test Starting... ")
# SentimentAnalysis() Method Test
# create a stub (client)
stub = grpc_services.SentimentAnalysisStub(channel)
# create a valid request message
test_data = b64_sentences.senteces()
message = rpc.InputMessage(value=test_data)
# make the call
response = stub.Analyze(message)
logger.debug("call => Analyze() Method Test Passed => " + response.value)
print()
except KeyError as e:
print(e)
path_setup.clean_paths()