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)