HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Check YouTube URL before tests (#12952)

Browse source
This commit is contained in:
Glenn Jocher 2024-05-20 20:45:11 +02:00 • committed by GitHub
parent 68031133fd
commit c8b6a8bc37
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_python.py
View file

@ -25,9 +25,8 @@ from ultralytics.utils import (
Retry, Retry,
checks, checks,
) )
from ultralytics.utils.downloads import download from ultralytics.utils.downloads import download, is_url
from ultralytics.utils.torch_utils import TORCH_1_9 from ultralytics.utils.torch_utils import TORCH_1_9
from . import CFG, IS_TMP_WRITEABLE, MODEL, SOURCE, TMP from . import CFG, IS_TMP_WRITEABLE, MODEL, SOURCE, TMP
@ -102,7 +101,7 @@ def test_predict_img(model_name):
def test_predict_visualize(model): def test_predict_visualize(model):
"""Test model predict methods with 'visualize=True' arguments.""" """Test model predict methods with 'visualize=True' arguments."""
YOLO(WEIGHTS_DIR / model)(SOURCE, imgsz=32, visualize=True) YOLO(WEIGHTS_DIR / model)(SOURCE, imgsz=32, visualize=True)
def test_predict_grey_and_4ch(): def test_predict_grey_and_4ch():
"""Test YOLO prediction on SOURCE converted to greyscale and 4-channel images.""" """Test YOLO prediction on SOURCE converted to greyscale and 4-channel images."""
@ -131,6 +130,7 @@ def test_predict_grey_and_4ch():
@pytest.mark.slow @pytest.mark.slow
@pytest.mark.skipif(not ONLINE, reason="environment is offline") @pytest.mark.skipif(not ONLINE, reason="environment is offline")
@pytest.mark.skipif(not is_url("https://youtu.be/G17sBkb38XQ"), reason="YouTube URL issue")
@Retry(times=3, delay=10) @Retry(times=3, delay=10)
def test_youtube(): def test_youtube():
""" """