class TestMethodError(TestCase):
def setUp(self):
self.aclient = Mock()
self.aclient.get_access_token = Mock(return_value='atoken')
self.aclient.get_api_key = Mock(return_value='apikey')
self.aclient.session = Mock(return_value={'access_token': 'atoken'})
self.vrs = VisualRecognition(self.aclient)
def test_add_face_file_not_found(self):
with pytest.raises(ValueError) as excinfo:
self.vrs.add_face('collection_id', 'image.jpg')
assert util.RESOURCE_ERROR == str(excinfo.value)
def test_add_face_file_not_found(self):
with pytest.raises(ValueError) as excinfo:
self.vrs.add_face('collection_id', 'image.jpg')
assert util.RESOURCE_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
def test_compare_faces_uuid_jpeg(self, opn, isfile, pil_open):
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.side_effect = [False, True]
img = MagicMock()
img.format.lower.return_value = 'jpeg'
pil_open.return_value = img
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('ef0dce93-c2ac-4da5-bb2c-82ca7c770ad8',
'test_1.jpeg')
assert util.COMBINATION_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
def test_compare_faces_uuid_uri(self, opn, isfile, pil_open):
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.side_effect = [False, False]
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('ef0dce93-c2ac-4da5-bb2c-82ca7c770ad8',
'https://test.co,/test.jpg')
assert util.COMBINATION_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
def test_compare_faces_uuid_uuid(self, opn, isfile, pil_open):
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.side_effect = [False, False]
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('ef0dce93-c2ac-4da5-bb2c-82ca7c770ad8',
'06ef969b-4d2f-49bf-8f79-afc3bc072def')
assert util.COMBINATION_ERROR == str(excinfo.value)
class TestMethodOK(TestCase):
def setUp(self):
self.aclient = Mock()
self.aclient.get_access_token = Mock(return_value='atoken')
self.aclient.get_api_key = Mock(return_value='apikey')
self.aclient.session = Mock(return_value={'access_token': 'atoken'})
self.vrs = VisualRecognition(self.aclient)
self.__create_face_collection_expected = {
'collection_id': '728bee35-fa67-473b-91bf-79f088f46179'
}
self.__list_face_collections_expected = {
'face_collections': [{
'collection_id':
'728bee35-fa67-473b-91bf-79f088f46179'
}]
}
self.__list_faces_expected = {
'faces': [{
'face_id': '728bee35-fa67-473b-91bf-79f088f46179'
}]
}
self.__add_face_expected = {
'face_id': '728bee35-fa67-473b-91bf-79f088f46179',
'location': {
'left': 1085,
'top': 244,
'right': 1307,
'bottom': 466
}
}
self.__compare_face_to_collection_expected = {
'source': {
'location': {
'left': 1085,
'top': 244,
'right': 1307,
'bottom': 466
}
},
'target': {
'collection_id':
'80bf2bdc-d3de-491e-9106-0635df0a0a18',
'faces': [{
'face_id': '80bf2bdc-d3de-491e-9106-0635df0a0a18',
'score': 0.787753701210022,
}]
}
}
def test_param_err(self):
with pytest.raises(TypeError):
self.vrs.detect_faces()
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_list_collection(self, req):
req.return_value.text = json.dumps(
self.__list_face_collections_expected)
req.return_value.status_code = 200
assert self.__list_face_collections_expected == self.vrs.list_collections(
)
headers = make_headers('application/json')
req.assert_called_once_with('GET',
ENDPOINT + '/face_collections',
headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_create_collection(self, req):
req.return_value.text = json.dumps(
self.__create_face_collection_expected)
req.return_value.status_code = 201
assert self.__create_face_collection_expected == self.vrs.create_collection(
)
headers = make_headers('application/json')
req.assert_called_once_with('POST',
ENDPOINT + '/face_collections',
headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_delete_collection(self, req):
req.return_value.text = ''
req.return_value.status_code = 204
assert '' == self.vrs.delete_collection(
'728bee35-fa67-473b-91bf-79f088f46179')
headers = make_headers('application/json')
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179'
req.assert_called_once_with('DELETE', ENDPOINT + uri, headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_list_faces(self, req):
req.return_value.text = json.dumps(self.__list_faces_expected)
req.return_value.status_code = 200
assert self.__list_faces_expected == self.vrs.list_faces(
'728bee35-fa67-473b-91bf-79f088f46179')
headers = make_headers('application/json')
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces'
req.assert_called_once_with('GET', ENDPOINT + uri, headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_remove_face(self, req):
req.return_value.text = ''
req.return_value.status_code = 204
assert '' == self.vrs.remove_face(
'728bee35-fa67-473b-91bf-79f088f46179',
'79a68ab3-8c42-4c79-bc09-3ac363cd9ab1')
headers = make_headers('application/json')
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces/79a68ab3-8c42-4c79-bc09-3ac363cd9ab1'
req.assert_called_once_with('DELETE', ENDPOINT + uri, headers=headers)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_add_face_jpeg(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__add_face_expected)
req.return_value.status_code = 201
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.side_effect = ['jpeg']
pil_open.return_value = img
assert self.__add_face_expected == self.vrs.add_face(
'test.jpg', '728bee35-fa67-473b-91bf-79f088f46179')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces'
payload = {'image': opn()}
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
files=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_add_face_png(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__add_face_expected)
req.return_value.status_code = 201
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.side_effect = ['png']
pil_open.return_value = img
assert self.__add_face_expected == self.vrs.add_face(
'test.png', '728bee35-fa67-473b-91bf-79f088f46179')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces'
payload = {'image': opn()}
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
files=payload)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_add_face_uri(self, req):
req.return_value.text = json.dumps(self.__add_face_expected)
req.return_value.status_code = 200
assert self.__add_face_expected == self.vrs.add_face(
'http://test.com/test.jpg', '728bee35-fa67-473b-91bf-79f088f46179')
headers = make_headers('application/json')
payload = json.dumps({'image': 'http://test.com/test.jpg'})
uri = '/face_collections/728bee35-fa67-473b-91bf-79f088f46179/faces'
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
data=payload)
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_uri_to_collection(self, req):
req.return_value.text = json.dumps(
self.__compare_face_to_collection_expected)
req.return_value.status_code = 200
assert self.__compare_face_to_collection_expected == self.vrs.compare_faces(
'http://test.com/test_1.jpg',
'728bee35-fa67-473b-91bf-79f088f46179')
headers = make_headers('application/json')
payload = json.dumps({'image': 'http://test.com/test_1.jpg'})
uri = '/compare_faces/728bee35-fa67-473b-91bf-79f088f46179'
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
data=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_image_to_collection(self, req, opn, isfile,
pil_open):
req.return_value.text = json.dumps(
self.__compare_face_to_collection_expected)
req.return_value.status_code = 200
isfile.side_effect = [True, False]
img = MagicMock()
img.format.lower.side_effect = ['jpeg']
pil_open.return_value = img
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
assert self.__compare_face_to_collection_expected == self.vrs.compare_faces(
'test_1.jpg', '728bee35-fa67-473b-91bf-79f088f46179')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
payload = {'image': opn()}
uri = '/compare_faces/728bee35-fa67-473b-91bf-79f088f46179'
req.assert_called_once_with('POST',
ENDPOINT + uri,
headers=headers,
files=payload)
class TestMethodOK(TestCase):
def setUp(self):
self.aclient = Mock()
self.aclient.get_access_token = Mock(return_value='atoken')
self.aclient.get_api_key = Mock(return_value='apikey')
self.aclient.session = Mock(return_value={'access_token': 'atoken'})
self.vrs = VisualRecognition(self.aclient)
self.__hd_expected = {
'humans': [{
'score': 1,
'location': {
'top': 100,
'right': 200,
'bottom': 200,
'left': 100
}
}]
}
self.__fd_expected = {
'faces': [{
'direction': 'left',
'location': {
'top': 100,
'right': 200,
'bottom': 200,
'left': 100
}
}]
}
self.__fr_expected = {
"score": 0.787753701210022,
"source": {
"location": {
"left": 1085,
"top": 244,
"right": 1307,
"bottom": 466
}
},
"target": {
"location": {
"left": 659,
"top": 207,
"right": 812,
"bottom": 360
}
}
}
def test_param_err(self):
with pytest.raises(TypeError):
self.vrs.detect_faces()
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_faces_uri(self, req):
req.return_value.text = json.dumps(self.__fd_expected)
req.return_value.status_code = 200
assert self.__fd_expected == self.vrs.detect_faces(
'http://test.com/test.jpg')
headers = make_headers('application/json')
payload = json.dumps({'image': 'http://test.com/test.jpg'})
req.assert_called_once_with('POST',
ENDPOINT + '/detect_faces',
headers=headers,
data=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_faces_jpeg(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__fd_expected)
req.return_value.status_code = 200
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.return_value = 'jpeg'
pil_open.return_value = img
assert self.__fd_expected == self.vrs.detect_faces('test.jpg')
isfile.assert_called_once_with('test.jpg')
pil_open.assert_called_once_with('test.jpg')
opn.assert_called_once_with('test.jpg', 'rb')
headers = make_headers('image/jpeg')
req.assert_called_once_with('POST',
ENDPOINT + '/detect_faces',
headers=headers,
data=opn())
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_faces_png(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__fd_expected)
req.return_value.status_code = 200
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.return_value = 'png'
pil_open.return_value = img
assert self.__fd_expected == self.vrs.detect_faces('test.png')
isfile.assert_called_once_with('test.png')
pil_open.assert_called_once_with('test.png')
opn.assert_called_once_with('test.png', 'rb')
headers = make_headers('image/png')
req.assert_called_once_with('POST',
ENDPOINT + '/detect_faces',
headers=headers,
data=opn())
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_humans_uri(self, req):
req.return_value.text = json.dumps(self.__hd_expected)
req.return_value.status_code = 200
assert self.__hd_expected == self.vrs.detect_humans(
'http://test.com/test.jpg')
headers = make_headers('application/json')
payload = json.dumps({'image': 'http://test.com/test.jpg'})
req.assert_called_once_with('POST',
ENDPOINT + '/detect_humans',
headers=headers,
data=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_humans_jpeg(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__hd_expected)
req.return_value.status_code = 200
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.return_value = 'jpeg'
pil_open.return_value = img
assert self.__hd_expected == self.vrs.detect_humans('test.jpg')
isfile.assert_called_once_with('test.jpg')
pil_open.assert_called_once_with('test.jpg')
opn.assert_called_once_with('test.jpg', 'rb')
headers = make_headers('image/jpeg')
req.assert_called_once_with('POST',
ENDPOINT + '/detect_humans',
headers=headers,
data=opn())
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_humans_png(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__hd_expected)
req.return_value.status_code = 200
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.return_value = 'png'
pil_open.return_value = img
assert self.__hd_expected == self.vrs.detect_humans('test.png')
isfile.assert_called_once_with('test.png')
pil_open.assert_called_once_with('test.png')
opn.assert_called_once_with('test.png', 'rb')
headers = make_headers('image/png')
req.assert_called_once_with('POST',
ENDPOINT + '/detect_humans',
headers=headers,
data=opn())
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_uri(self, req):
req.return_value.text = json.dumps(self.__fr_expected)
req.return_value.status_code = 200
assert self.__fr_expected == self.vrs.compare_faces(
'http://test.com/test_1.jpg', 'http://test.com/test_2.jpg')
headers = make_headers('application/json')
payload = json.dumps({
'source': 'http://test.com/test_1.jpg',
'target': 'http://test.com/test_2.jpg'
})
req.assert_called_once_with('POST',
ENDPOINT + '/compare_faces',
headers=headers,
data=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_jpeg_jpeg(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__fr_expected)
req.return_value.status_code = 200
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.side_effect = ['jpeg', 'jpeg']
pil_open.return_value = img
assert self.__fr_expected == self.vrs.compare_faces(
'test_1.jpg', 'test_2.jpg')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
payload = {'source': opn(), 'target': opn()}
req.assert_called_once_with('POST',
ENDPOINT + '/compare_faces',
headers=headers,
files=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_png_png(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__fr_expected)
req.return_value.status_code = 200
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.side_effect = ['png', 'png']
pil_open.return_value = img
assert self.__fr_expected == self.vrs.compare_faces(
'test_1.png', 'test_2.png')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
payload = {'source': opn(), 'target': opn()}
req.assert_called_once_with('POST',
ENDPOINT + '/compare_faces',
headers=headers,
files=payload)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_jpeg_png(self, req, opn, isfile, pil_open):
req.return_value.text = json.dumps(self.__fr_expected)
req.return_value.status_code = 200
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.side_effect = ['jpeg', 'png']
pil_open.return_value = img
assert self.__fr_expected == self.vrs.compare_faces(
'test_1.jpeg', 'test_2.png')
headers = {'Authorization': 'Bearer atoken', 'x-api-key': 'apikey'}
payload = {'source': opn(), 'target': opn()}
req.assert_called_once_with('POST',
ENDPOINT + '/compare_faces',
headers=headers,
files=payload)
class TestMethodError(TestCase):
def setUp(self):
self.aclient = Mock()
self.aclient.get_access_token = Mock(return_value='atoken')
self.aclient.get_api_key = Mock(return_value='apikey')
self.aclient.session = Mock(return_value={'access_token': 'atoken'})
self.vrs = VisualRecognition(self.aclient)
def test_detect_humans_file_not_found(self):
with pytest.raises(ValueError) as excinfo:
self.vrs.detect_humans('image.jpg')
assert util.RESOURCE_ERROR == str(excinfo.value)
def test_detect_faces_file_not_found(self):
with pytest.raises(ValueError) as excinfo:
self.vrs.detect_faces('image.jpg')
assert util.RESOURCE_ERROR == str(excinfo.value)
def test_compare_faces_file_not_found(self):
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('image_1.jpg', 'image_2.jpg')
assert util.RESOURCE_ERROR == str(excinfo.value)
@mock.patch('json.loads')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_humans_resource_not_found(self, req, ret):
req.return_value.text = 'test'
req.return_value.status_code = 404
ret.return_value = {
'error': {
'code': 'resource_not_found',
'message': 'The specified resource does not exist.'
}
}
try:
self.vrs.detect_humans('http://test.com/test.jpg')
except ClientError as excinfo:
assert ret.return_value == excinfo.response
assert req.return_value.status_code == excinfo.status_code
@mock.patch('json.loads')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_faces_resource_not_found(self, req, ret):
req.return_value.text = 'test'
req.return_value.status_code = 404
ret.return_value = {
'error': {
'code': 'resource_not_found',
'message': 'The specified resource does not exist.'
}
}
try:
self.vrs.detect_faces('http://test.com/test.jpg')
except ClientError as excinfo:
assert ret.return_value == excinfo.response
assert req.return_value.status_code == excinfo.status_code
@mock.patch('json.loads')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_resource_not_found(self, req, ret):
req.return_value.text = 'test'
req.return_value.status_code = 404
ret.return_value = {
'error': {
'code': 'resource_not_found',
'message': 'The specified resource does not exist.'
}
}
try:
self.vrs.compare_faces('http://test.com/test_1.jpg',
'http://test.com/test_2.jpg')
except ClientError as excinfo:
assert ret.return_value == excinfo.response
assert req.return_value.status_code == excinfo.status_code
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_humans_gif(self, req, opn, isfile, pil_open):
req.side_effect = RequestException
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.return_value = 'gif'
pil_open.return_value = img
with pytest.raises(ValueError) as excinfo:
self.vrs.detect_humans('image.gif')
assert util.UNSUPPORTED_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_detect_faces_gif(self, req, opn, isfile, pil_open):
req.side_effect = RequestException
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.return_value = 'gif'
pil_open.return_value = img
with pytest.raises(ValueError) as excinfo:
self.vrs.detect_faces('image.gif')
assert util.UNSUPPORTED_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
@mock.patch('ricohcloudsdk.vrs.util.SESSION.request')
def test_compare_faces_gif(self, req, opn, isfile, pil_open):
req.side_effect = RequestException
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.return_value = True
img = MagicMock()
img.format.lower.return_value = 'gif'
pil_open.return_value = img
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('image_1.gif', 'image_2.gif')
assert util.UNSUPPORTED_ERROR == str(excinfo.value)
@mock.patch('ricohcloudsdk.vrs.util.Image.open')
@mock.patch('os.path.isfile')
@mock.patch('ricohcloudsdk.vrs.client.open')
def test_compare_faces_jpeg_uri(self, opn, isfile, pil_open):
opn.side_effect = mock.mock_open()
opn.read_data = b'readdata'
isfile.side_effect = [True, False]
img = MagicMock()
img.format.lower.return_value = 'jpeg'
pil_open.return_value = img
with pytest.raises(ValueError) as excinfo:
self.vrs.compare_faces('test_1.jpeg', 'https://test.co,/test.jpg')
assert util.COMBINATION_ERROR == str(excinfo.value)