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)
def __init__(self):
""" read config file and initialize auth client """
with open('./config.json', 'r') as settings:
config = json.load(settings)
client_id = config['CLIENT_ID']
client_secret = config['CLIENT_SECRET']
vrs_auth_client = AuthClient(client_id, client_secret)
self.vrs_client = VisualRecognition(vrs_auth_client)
ips_auth_client = AuthClient(client_id, client_secret)
self.ips_client = ImageProcessing(ips_auth_client)
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 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 create_vrs_client():
"""Read config file and creates client for visual recognition SDK."""
with open('./config.json', 'r') as settings:
config = json.load(settings)
client_id = config['CLIENT_ID']
client_secret = config['CLIENT_SECRET']
auth_client = AuthClient(client_id, client_secret)
vrs_client = VisualRecognition(auth_client)
return vrs_client
class HumanDetectionAndBlurFilterSample(object):
""" human detection and blur filter sample """
def __init__(self):
""" read config file and initialize auth client """
with open('./config.json', 'r') as settings:
config = json.load(settings)
client_id = config['CLIENT_ID']
client_secret = config['CLIENT_SECRET']
vrs_auth_client = AuthClient(client_id, client_secret)
self.vrs_client = VisualRecognition(vrs_auth_client)
ips_auth_client = AuthClient(client_id, client_secret)
self.ips_client = ImageProcessing(ips_auth_client)
def __detect_humans(self, img_path):
res = self.vrs_client.detect_humans(img_path)
return [human['location'] for human in res['humans']]
def __blur_filter(self, img_path, options, locations):
parameters = {'locations': locations, 'type': 'blur', 'options': options}
res = self.ips_client.filter(img_path, parameters)
return res
def main(self):
""" main """
parser = argparse.ArgumentParser(description='Blur filter sample')
parser.add_argument('-f', '--file', type=str, dest='file_path', help='specify image file (JPEG or PNG) to process')
parser.add_argument('--ksize_width', type=int, dest='ksize_width', default=31, help='specify kernel size width')
parser.add_argument('--ksize_height', type=int, dest='ksize_height', default=31, help='specify kernel size height')
args = parser.parse_args()
if args.file_path is None:
parser.print_help()
return
locations = self.__detect_humans(args.file_path)
options = {
'locations': {
'shape': 'min_enclosing_circle',
'edge': 'blur'
},
'ksize_width': args.ksize_width,
'ksize_height': args.ksize_height,
}
res = self.__blur_filter(args.file_path, options, locations)
if not os.path.exists('./results'):
os.mkdir('./results')
with open('./results/human_detection_and_blur_filter.jpg', 'wb') as f:
f.write(res)
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)
def test_param_err(self):
with pytest.raises(TypeError):
VisualRecognition()
def test_ok(self):
VisualRecognition(self.aclient)
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)
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)
def create_vr_client(profile):
"""This function create Visual Recognition Service object."""
credential = get_credential_info(profile)
auth_client = AuthClient(credential[CLIENT_ID], credential[CLIENT_SECRET])
vr_client = VisualRecognition(auth_client)
return vr_client
