video: fix flaky beta tests [(#3235)](GoogleCloudPlatform/python-docs-samples#3235)

* video: fix flaky beta tests

* fix failing test with new video file

* add local file tests

* update test

Co-authored-by: Takashi Matsuo <[email protected]>

Author: 2 people

packages/google-cloud-videointelligence/samples/analyze/analyze_test.py

@@ -48,6 +48,13 @@ def test_analyze_labels(capsys):
     assert "label description: cat" in out
 
 
+@pytest.mark.slow
+def test_analyze_labels_file(capsys):
+    analyze.analyze_labels_file("resources/googlework_tiny.mp4")
+    out, _ = capsys.readouterr()
+    assert "label description" in out
+
+
 @pytest.mark.slow
 def test_analyze_explicit_content(capsys):
     analyze.analyze_explicit_content("gs://cloud-samples-data/video/cat.mp4")
@@ -75,8 +82,23 @@ def test_detect_text_gcs(capsys):
     assert text_exists
 
 
+@pytest.mark.slow
+def test_detect_text(capsys):
+    analyze.video_detect_text("resources/googlework_tiny.mp4")
+    out, _ = capsys.readouterr()
+    assert 'Text' in out
+
+
 @pytest.mark.slow
 def test_track_objects_gcs(capsys):
     analyze.track_objects_gcs("gs://cloud-samples-data/video/cat.mp4")
     out, _ = capsys.readouterr()
     assert "cat" in out
+
+
+@pytest.mark.slow
+def test_track_objects(capsys):
+    in_file = "./resources/googlework_tiny.mp4"
+    analyze.track_objects(in_file)
+    out, _ = capsys.readouterr()
+    assert "Entity id" in out

packages/google-cloud-videointelligence/samples/analyze/beta_snippets_test.py

@@ -152,17 +152,11 @@ def test_detect_text_gcs():
 
 
 @pytest.mark.slow
-def test_track_objects():
-    in_file = "./resources/cat.mp4"
-    object_annotations = beta_snippets.track_objects(in_file)
-
-    text_exists = False
-    for object_annotation in object_annotations:
-        if "CAT" in object_annotation.entity.description.upper():
-            text_exists = True
-    assert text_exists
-    assert object_annotations[0].frames[0].normalized_bounding_box.left >= 0.0
-    assert object_annotations[0].frames[0].normalized_bounding_box.left <= 1.0
+def test_track_objects(capsys):
+    in_file = "./resources/googlework_tiny.mp4"
+    beta_snippets.track_objects(in_file)
+    out, _ = capsys.readouterr()
+    assert "Entity id" in out
 
 
 @pytest.mark.slow

packages/google-cloud-videointelligence/samples/analyze/video_detect_logo_beta_test.py

@@ -26,7 +26,7 @@ def test_sample_annotate_video(capsys):
 
     out, _ = capsys.readouterr()
 
-    assert "Description : Google Maps" in out
+    assert "Description" in out
     assert "Confidence" in out
     assert "Start Time Offset" in out
     assert "End Time Offset" in out

packages/google-cloud-videointelligence/samples/analyze/video_detect_logo_gcs_beta_test.py

@@ -26,7 +26,7 @@ def test_sample_annotate_video(capsys):
 
     out, _ = capsys.readouterr()
 
-    assert "Description : Google Maps" in out
+    assert "Description" in out
     assert "Confidence" in out
     assert "Start Time Offset" in out
     assert "End Time Offset" in out