richkadel
diff --git a/‎system_tests/data/car.jpg‎
65.5 KB b/‎system_tests/data/car.jpg‎
65.5 KB
diff --git a/‎system_tests/vision.py‎
Lines changed: 74 additions & 7 deletions b/‎system_tests/vision.py‎
Lines changed: 74 additions & 7 deletions
@@ -24,11 +24,13 @@
 
 from system_test_utils import unique_resource_id
 from retry import RetryErrors
+import six
 
 
 _SYS_TESTS_DIR = os.path.abspath(os.path.dirname(__file__))
 LOGO_FILE = os.path.join(_SYS_TESTS_DIR, 'data', 'logo.png')
 FACE_FILE = os.path.join(_SYS_TESTS_DIR, 'data', 'faces.jpg')
+LABEL_FILE = os.path.join(_SYS_TESTS_DIR, 'data', 'car.jpg')
 
 
 class Config(object):
@@ -54,6 +56,14 @@ def tearDownModule():
     bucket_retry(Config.TEST_BUCKET.delete)(force=True)
 
 
+class BaseVisionTestCase(unittest.TestCase):
+    def _assert_coordinate(self, coordinate):
+        if coordinate is None:
+            return
+        self.assertIsInstance(coordinate, (int, float))
+        self.assertNotEqual(coordinate, 0.0)
+
+
 class TestVisionClientLogo(unittest.TestCase):
     def setUp(self):
         self.to_delete_by_case = []
@@ -111,20 +121,14 @@ def test_detect_logos_gcs(self):
         self._assert_logo(logo)
 
 
-class TestVisionClientFace(unittest.TestCase):
+class TestVisionClientFace(BaseVisionTestCase):
     def setUp(self):
         self.to_delete_by_case = []
 
     def tearDown(self):
         for value in self.to_delete_by_case:
             value.delete()
 
-    def _assert_coordinate(self, coordinate):
-        if coordinate is None:
-            return
-        self.assertIsInstance(coordinate, (int, float))
-        self.assertGreater(abs(coordinate), 0.0)
-
     def _assert_likelihood(self, likelihood):
         from google.cloud.vision.likelihood import Likelihood
 
@@ -215,3 +219,66 @@ def test_detect_faces_filename(self):
         self.assertEqual(len(faces), 5)
         for face in faces:
             self._assert_face(face)
+
+
+class TestVisionClientLabel(BaseVisionTestCase):
+    DESCRIPTIONS = (
+        'car',
+        'vehicle',
+        'land vehicle',
+        'automotive design',
+        'wheel',
+        'automobile make',
+        'luxury vehicle',
+        'sports car',
+        'performance car',
+        'automotive exterior',
+    )
+
+    def setUp(self):
+        self.to_delete_by_case = []
+
+    def tearDown(self):
+        for value in self.to_delete_by_case:
+            value.delete()
+
+    def _assert_label(self, label):
+
+        self.assertIsInstance(label, EntityAnnotation)
+        self.assertIn(label.description, self.DESCRIPTIONS)
+        self.assertIsInstance(label.mid, six.text_type)
+        self.assertGreater(label.score, 0.0)
+
+    def test_detect_labels_content(self):
+        client = Config.CLIENT
+        with open(LABEL_FILE, 'rb') as image_file:
+            image = client.image(content=image_file.read())
+        labels = image.detect_labels()
+        self.assertEqual(len(labels), 10)
+        for label in labels:
+            self._assert_label(label)
+
+    def test_detect_labels_gcs(self):
+        bucket_name = Config.TEST_BUCKET.name
+        blob_name = 'car.jpg'
+        blob = Config.TEST_BUCKET.blob(blob_name)
+        self.to_delete_by_case.append(blob)  # Clean-up.
+        with open(LABEL_FILE, 'rb') as file_obj:
+            blob.upload_from_file(file_obj)
+
+        source_uri = 'gs://%s/%s' % (bucket_name, blob_name)
+
+        client = Config.CLIENT
+        image = client.image(source_uri=source_uri)
+        labels = image.detect_labels()
+        self.assertEqual(len(labels), 10)
+        for label in labels:
+            self._assert_label(label)
+
+    def test_detect_labels_filename(self):
+        client = Config.CLIENT
+        image = client.image(filename=LABEL_FILE)
+        labels = image.detect_labels()
+        self.assertEqual(len(labels), 10)
+        for label in labels:
+            self._assert_label(label)