chore(fixit): sample clean up for vision quickstart (GoogleCloudPlatform#10050)

balajismaniam · web-flow · commit f29a06536590 · 2023-05-24T13:15:29.000-07:00
* chore(fixit): sample clean up for vision quickstart

* added back printing labels

---------

Co-authored-by: Balaji Subramaniam &lt;balajismaniam@google.com&gt;
diff --git a/vision/snippets/quickstart/quickstart.py b/vision/snippets/quickstart/quickstart.py
@@ -14,26 +14,29 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+# [START vision_quickstart]
+import os
 
-def run_quickstart():
-    # [START vision_quickstart]
-    import os
+# Imports the Google Cloud client library
+# [START vision_python_migration_import]
+from google.cloud import vision
 
-    # Imports the Google Cloud client library
-    # [START vision_python_migration_import]
-    from google.cloud import vision
-    # [END vision_python_migration_import]
+# [END vision_python_migration_import]
+
+
+def run_quickstart() -> vision.EntityAnnotation:
+    """Provides a quick start example for Cloud Vision."""
 
     # Instantiates a client
     # [START vision_python_migration_client]
     client = vision.ImageAnnotatorClient()
     # [END vision_python_migration_client]
 
     # The name of the image file to annotate
-    file_name = os.path.abspath('resources/wakeupcat.jpg')
+    file_name = os.path.abspath("resources/wakeupcat.jpg")
 
     # Loads the image into memory
-    with open(file_name, 'rb') as image_file:
+    with open(file_name, "rb") as image_file:
         content = image_file.read()
 
     image = vision.Image(content=content)
@@ -45,8 +48,11 @@ def run_quickstart():
     print('Labels:')
     for label in labels:
         print(label.description)
-    # [END vision_quickstart]
 
+    return labels
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     run_quickstart()
+
+# [END vision_quickstart]
diff --git a/vision/snippets/quickstart/quickstart_test.py b/vision/snippets/quickstart/quickstart_test.py
@@ -15,7 +15,7 @@
 import quickstart
 
 
-def test_quickstart(capsys):
-    quickstart.run_quickstart()
-    out, _ = capsys.readouterr()
-    assert 'Labels' in out
+def test_quickstart() -> None:
+    labels = quickstart.run_quickstart()
+    for label in labels:
+        assert len(label.description) > 0
