class TestSingleImageHelper(unittest.TestCase):
def setUp(self):
credentials = mock.Mock(spec=Credentials)
self.client = ImageAnnotatorClient(credentials=credentials)
@mock.patch.object(ImageAnnotatorClient, 'batch_annotate_images')
def test_all_features_default(self, batch_annotate):
# Set up an image annotation request with no features.
image = types.Image(source={
'image_uri': 'http://foo.com/img.jpg',
})
request = types.AnnotateImageRequest(image=image)
assert not request.features
# Perform the single image request.
self.client.annotate_image(request)
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(args[0]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = args[0][0]
all_features = self.client._get_all_features()
assert request_sent.image is request.image
assert len(request_sent.features) == len(all_features)
@mock.patch.object(ImageAnnotatorClient, 'batch_annotate_images')
def test_explicit_features(self, batch_annotate):
# Set up an image annotation request with no features.
image = types.Image(source={
'image_uri': 'http://foo.com/img.jpg',
})
request = types.AnnotateImageRequest(
image=image,
features=[
types.Feature(type=1),
types.Feature(type=2),
types.Feature(type=3),
],
)
# Perform the single image request.
self.client.annotate_image(request)
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(args[0]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = args[0][0]
assert request_sent.image is request.image
assert len(request_sent.features) == 3
for feature, i in zip(request_sent.features, range(1, 4)):
assert feature.type == i
assert feature.max_results == 0
@mock.patch.object(ImageAnnotatorClient, 'batch_annotate_images')
def test_image_file_handler(self, batch_annotate):
# Set up a file handler.
file_ = io.BytesIO(b'bogus==')
# Perform the single image request.
self.client.annotate_image({'image': file_})
# Evaluate the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(args[0]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = args[0][0]
assert request_sent['image']['content'] == b'bogus=='
@mock.patch.object(ImageAnnotatorClient, 'batch_annotate_images')
@mock.patch.object(io, 'open')
def test_image_filename(self, io_open, batch_annotate):
# Make io.open send back a mock with a read method.
file_ = mock.MagicMock(spec=io.BytesIO)
io_open.return_value = file_
file_.__enter__.return_value = file_
file_.read.return_value = b'imagefile=='
# Perform the single image request using a filename.
self.client.annotate_image(
{'image': {
'source': {
'filename': 'image.jpeg'
}
}}, )
# Establish that my file was opened.
io_open.assert_called_once_with('image.jpeg', 'rb')
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(args[0]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = args[0][0]
assert request_sent['image']['content'] == b'imagefile=='
content = image_file.read()
texts = client.document_text_detection(imageData, content=content)
print(texts[0].description)
#############################
from google.cloud.vision_v1 import ImageAnnotatorClient
client = ImageAnnotatorClient()
request = {
'image': {
'source': {
'image_uri': 'https://foo.com/image.jpg'
},
},
}
response = client.annotate_image(request)
###############
#import requests
#import base64
#
#file_name = '/sss.jpg'
#with open(file_name, 'r') as image:
# image_content = image.read()
# encoded_content = base64.b64encode(image_content)
#
#data = {
# "requests":[
# {
# "image":{
class TestSingleImageHelper(unittest.TestCase):
def setUp(self):
credentials = mock.Mock(spec=Credentials)
self.client = ImageAnnotatorClient(credentials=credentials)
@mock.patch.object(ImageAnnotatorClient, "batch_annotate_images")
def test_all_features_default(self, batch_annotate):
# Set up an image annotation request with no features.
image = vision_v1.Image(source={"image_uri": "http://foo.com/img.jpg"})
request = vision_v1.AnnotateImageRequest(image=image)
assert not request.features
# Perform the single image request.
self.client.annotate_image(request)
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(kwargs["requests"]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = kwargs["requests"][0]
all_features = self.client._get_all_features()
assert request_sent.image == request.image
assert len(request_sent.features) == len(all_features)
@mock.patch.object(ImageAnnotatorClient, "batch_annotate_images")
def test_explicit_features(self, batch_annotate):
# Set up an image annotation request with no features.
image = vision_v1.Image(source={"image_uri": "http://foo.com/img.jpg"})
request = vision_v1.AnnotateImageRequest(
image=image,
features=[
vision_v1.Feature(type_=1),
vision_v1.Feature(type_=2),
vision_v1.Feature(type_=3),
],
)
# Perform the single image request.
self.client.annotate_image(request)
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(kwargs["requests"]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = kwargs["requests"][0]
assert request_sent.image == request.image
assert len(request_sent.features) == 3
for feature, i in zip(request_sent.features, range(1, 4)):
assert feature.type_ == i
assert feature.max_results == 0
@mock.patch.object(ImageAnnotatorClient, "batch_annotate_images")
def test_image_file_handler(self, batch_annotate):
# Set up a file handler.
file_ = io.BytesIO(b"bogus==")
# Perform the single image request.
self.client.annotate_image({"image": file_})
# Evaluate the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(kwargs["requests"]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = kwargs["requests"][0]
assert request_sent["image"]["content"] == b"bogus=="
@mock.patch.object(ImageAnnotatorClient, "batch_annotate_images")
@mock.patch.object(builtins, "open")
def test_image_filename(self, io_open, batch_annotate):
# Make io.open send back a mock with a read method.
file_ = mock.MagicMock(spec=io.BytesIO)
io_open.return_value = file_
file_.__enter__.return_value = file_
file_.read.return_value = b"imagefile=="
# Perform the single image request using a filename.
self.client.annotate_image(
{"image": {
"source": {
"filename": "image.jpeg"
}
}})
# Establish that my file was opened.
io_open.assert_called_once_with("image.jpeg", "rb")
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(kwargs["requests"]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = kwargs["requests"][0]
assert request_sent["image"]["content"] == b"imagefile=="
class TestSingleImageHelper(unittest.TestCase):
def setUp(self):
credentials = mock.Mock(spec=Credentials)
self.client = ImageAnnotatorClient(credentials=credentials)
@mock.patch.object(ImageAnnotatorClient, "batch_annotate_images")
def test_all_features_default(self, batch_annotate):
# Set up an image annotation request with no features.
image = types.Image(source={"image_uri": "http://foo.com/img.jpg"})
request = types.AnnotateImageRequest(image=image)
assert not request.features
# Perform the single image request.
self.client.annotate_image(request)
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(args[0]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = args[0][0]
all_features = self.client._get_all_features()
assert request_sent.image is request.image
assert len(request_sent.features) == len(all_features)
@mock.patch.object(ImageAnnotatorClient, "batch_annotate_images")
def test_explicit_features(self, batch_annotate):
# Set up an image annotation request with no features.
image = types.Image(source={"image_uri": "http://foo.com/img.jpg"})
request = types.AnnotateImageRequest(
image=image,
features=[
types.Feature(type=1),
types.Feature(type=2),
types.Feature(type=3),
],
)
# Perform the single image request.
self.client.annotate_image(request)
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(args[0]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = args[0][0]
assert request_sent.image is request.image
assert len(request_sent.features) == 3
for feature, i in zip(request_sent.features, range(1, 4)):
assert feature.type == i
assert feature.max_results == 0
@mock.patch.object(ImageAnnotatorClient, "batch_annotate_images")
def test_image_file_handler(self, batch_annotate):
# Set up a file handler.
file_ = io.BytesIO(b"bogus==")
# Perform the single image request.
self.client.annotate_image({"image": file_})
# Evaluate the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(args[0]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = args[0][0]
assert request_sent["image"]["content"] == b"bogus=="
@mock.patch.object(ImageAnnotatorClient, "batch_annotate_images")
@mock.patch.object(io, "open")
def test_image_filename(self, io_open, batch_annotate):
# Make io.open send back a mock with a read method.
file_ = mock.MagicMock(spec=io.BytesIO)
io_open.return_value = file_
file_.__enter__.return_value = file_
file_.read.return_value = b"imagefile=="
# Perform the single image request using a filename.
self.client.annotate_image({"image": {"source": {"filename": "image.jpeg"}}})
# Establish that my file was opened.
io_open.assert_called_once_with("image.jpeg", "rb")
# Evalute the argument sent to batch_annotate_images.
assert batch_annotate.call_count == 1
_, args, kwargs = batch_annotate.mock_calls[0]
# Only a single request object should be sent.
assert len(args[0]) == 1
# Evalute the request object to ensure it looks correct.
request_sent = args[0][0]
assert request_sent["image"]["content"] == b"imagefile=="
