update samples for product search GA [(#1861)](GoogleCloudPlatform/py…

…thon-docs-samples#1861) * update samples for product search GA * update to use 0.35.1
busunkim96 · Nov 26, 2018 · ee1897b · ee1897b
1 parent 4d885a4
commit ee1897b
Show file tree

Hide file tree

Showing 8 changed files with 11 additions and 11 deletions.
diff --git a/samples/snippets/product_search/import_product_sets.py b/samples/snippets/product_search/import_product_sets.py
@@ -24,7 +24,7 @@
 import argparse
 
 # [START vision_product_search_tutorial_import]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 # [END vision_product_search_tutorial_import]
 
 

diff --git a/samples/snippets/product_search/product_in_product_set_management.py b/samples/snippets/product_search/product_in_product_set_management.py
@@ -25,7 +25,7 @@
 
 # [START vision_product_search_add_product_to_product_set]
 # [START vision_product_search_remove_product_from_product_set]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_add_product_to_product_set]
 # [END vision_product_search_remove_product_from_product_set]

diff --git a/samples/snippets/product_search/product_in_product_set_management_test.py b/samples/snippets/product_search/product_in_product_set_management_test.py
@@ -54,24 +54,24 @@ def product_and_product_set():
 def test_add_product_to_product_set(capsys, product_and_product_set):
     list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
     out, _ = capsys.readouterr()
-    assert PRODUCT_ID not in out
+    assert 'Product id: {}'.format(PRODUCT_ID) not in out
 
     add_product_to_product_set(
         PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID)
     list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
     out, _ = capsys.readouterr()
-    assert PRODUCT_ID in out
+    assert 'Product id: {}'.format(PRODUCT_ID) in out
 
 
 def test_remove_product_from_product_set(capsys, product_and_product_set):
     add_product_to_product_set(
         PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID)
     list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
     out, _ = capsys.readouterr()
-    assert PRODUCT_ID in out
+    assert 'Product id: {}'.format(PRODUCT_ID) in out
 
     remove_product_from_product_set(
         PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID)
     list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
     out, _ = capsys.readouterr()
-    assert PRODUCT_ID not in out
+    assert 'Product id: {}'.format(PRODUCT_ID) not in out
diff --git a/samples/snippets/product_search/product_management.py b/samples/snippets/product_search/product_management.py
@@ -28,7 +28,7 @@
 # [START vision_product_search_list_products]
 # [START vision_product_search_get_product]
 # [START vision_product_search_update_product_labels]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_create_product]
 # [END vision_product_search_delete_product]

diff --git a/samples/snippets/product_search/product_search.py b/samples/snippets/product_search/product_search.py
@@ -26,7 +26,7 @@
 
 # [START vision_product_search_get_similar_products]
 # [START vision_product_search_get_similar_products_gcs]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_get_similar_products]
 # [END vision_product_search_get_similar_products_gcs]

diff --git a/samples/snippets/product_search/product_set_management.py b/samples/snippets/product_search/product_set_management.py
@@ -27,7 +27,7 @@
 # [START vision_product_search_list_product_sets]
 # [START vision_product_search_get_product_set]
 # [START vision_product_search_create_product_set]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_delete_product_set]
 # [END vision_product_search_list_product_sets]

diff --git a/samples/snippets/product_search/reference_image_management.py b/samples/snippets/product_search/reference_image_management.py
@@ -27,7 +27,7 @@
 # [START vision_product_search_delete_reference_image]
 # [START vision_product_search_list_reference_images]
 # [START vision_product_search_get_reference_image]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_create_reference_image]
 # [END vision_product_search_delete_reference_image]

diff --git a/samples/snippets/product_search/requirements.txt b/samples/snippets/product_search/requirements.txt
@@ -1 +1 @@
-google-cloud-vision==0.35.0
+google-cloud-vision==0.35.1