diff --git a/video/cloud-client/analyze/analyze_test.py b/video/cloud-client/analyze/analyze_test.py index c66de346f08..108b329ecb8 100644 --- a/video/cloud-client/analyze/analyze_test.py +++ b/video/cloud-client/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 diff --git a/video/cloud-client/analyze/beta_snippets_test.py b/video/cloud-client/analyze/beta_snippets_test.py index 715a46bf978..84be8048c58 100644 --- a/video/cloud-client/analyze/beta_snippets_test.py +++ b/video/cloud-client/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 diff --git a/video/cloud-client/analyze/video_detect_logo_beta_test.py b/video/cloud-client/analyze/video_detect_logo_beta_test.py index 1605fc499c9..6c6dc00f5b8 100644 --- a/video/cloud-client/analyze/video_detect_logo_beta_test.py +++ b/video/cloud-client/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 diff --git a/video/cloud-client/analyze/video_detect_logo_gcs_beta_test.py b/video/cloud-client/analyze/video_detect_logo_gcs_beta_test.py index 7c6c6530327..1891886cf39 100644 --- a/video/cloud-client/analyze/video_detect_logo_gcs_beta_test.py +++ b/video/cloud-client/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