tests/pipeline/test_download_videos.py

from pipeline.download_videos import get_id


def test_get_id():
    url1 = "https://www.youtube.com/watch?v=frYIj2FGmMA&foo=bar"
    url2 = "https://www.youtube.com/watch?v=abcdefg"
    url3 = "https://www.youtube.com/watch?foo=bar&v=xyz123"
    assert get_id(url1) == "frYIj2FGmMA"
    assert get_id(url2) == "abcdefg"
    assert get_id(url3) == "xyz123"