def test_missing_words(self):
request = service_pb2.VectorsRequest(model_name="correct")
try:
self.stub.GetWordsVectors(request, None)
except grpc.RpcError as error:
self.assertTrue(
error._state.code == grpc.StatusCode.INVALID_ARGUMENT)
def test_missing_model(self):
request = service_pb2.VectorsRequest(batch=["price"])
try:
self.stub.GetWordsVectors(request, None)
except grpc.RpcError as error:
self.assertTrue(
error._state.code == grpc.StatusCode.INVALID_ARGUMENT)
def get_word_vector(self, *models):
response = []
for model in models:
request = service_pb2.VectorsRequest(model_name=model,
batch=["price"])
response += self.stub.GetWordsVectors(request, None).vectors
return response
def test_model_not_loaded(self):
request = service_pb2.VectorsRequest(model_name="fghfdhg",
batch=["price"])
try:
self.stub.GetWordsVectors(request, None)
except grpc.RpcError as error:
self.assertTrue(
error._state.code == grpc.StatusCode.FAILED_PRECONDITION)
def test_various_words(self):
request = service_pb2.VectorsRequest(model_name="correct",
batch=["price", "quantity"])
response = self.stub.GetWordsVectors(request, None)
self.assertTrue(len(list(response.vectors)) == 2)
self.assertTrue(response.model.name == request.model_name)