File size: 2,160 Bytes
a0ca1c2 49627f0 a0ca1c2 49627f0 328e22b 49627f0 328e22b 61b18c9 c7833ef 49627f0 f9618bd 49627f0 24c3a19 328e22b 24c3a19 328e22b 24c3a19 328e22b 7f9dcc9 f8326d6 7f9dcc9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
# -*- coding: utf-8 -*-
"""Unit tests for the :module:`extract <extract>` module."""
from pytube import extract
def test_extract_video_id():
url = 'https://www.youtube.com/watch?v=9bZkp7q19f0'
video_id = extract.video_id(url)
assert video_id == '9bZkp7q19f0'
def test_extract_watch_url():
video_id = '9bZkp7q19f0'
watch_url = extract.watch_url(video_id)
assert watch_url == 'https://youtube.com/watch?v=9bZkp7q19f0'
def test_info_url(cipher_signature):
video_info_url = extract.video_info_url(
video_id=cipher_signature.video_id,
watch_url=cipher_signature.watch_url,
watch_html=cipher_signature.watch_html,
embed_html='',
age_restricted=False,
)
expected = (
'https://youtube.com/get_video_info?video_id=9bZkp7q19f0&el=%24el'
'&ps=default&eurl=https%253A%2F%2Fyoutube.com%2Fwatch%253Fv%'
'253D9bZkp7q19f0&hl=en_US'
)
assert video_info_url == expected
def test_js_url(cipher_signature):
expected = 'https://youtube.com/yts/jsbin/player-vflOdyxa4/en_US/base.js'
result = extract.js_url(cipher_signature.watch_html)
assert expected == result
def test_age_restricted(age_restricted):
assert extract.is_age_restricted(age_restricted['watch_html'])
def test_non_age_restricted(cipher_signature):
assert not extract.is_age_restricted(cipher_signature.watch_html)
def test_get_vid_desc(cipher_signature):
expected = "PSY - DADDY(feat. CL of 2NE1) M/V @ https://youtu.be/FrG4TEcSuRg\nPSY - ๋ํ๋ฐ์ง(NAPAL BAJI) M/V @ https://youtu.be/tF27TNC_4pc\nPSY - 7TH ALBUM '์น ์ง์ธ์ด๋ค' on iTunes @ http://smarturl.it/PSY_7THALBUM\nPSY - GANGNAM STYLE(๊ฐ๋จ์คํ์ผ) on iTunes @ http://smarturl.it/PsyGangnam\n#PSY #์ธ์ด #GANGNAMSTYLE #๊ฐ๋จ์คํ์ผ\nMore about PSY@\nhttp://www.psypark.com/\nhttp://www.youtube.com/officialpsy\nhttp://www.facebook.com/officialpsy\nhttp://twitter.com/psy_oppa\nhttps://www.instagram.com/42psy42\nhttp://iTunes.com/PSY\nhttp://sptfy.com/PSY\nhttp://weibo.com/psyoppa\nhttp://twitter.com/ygent_official" # noqa
assert extract.get_vid_descr(cipher_signature.watch_html) == expected
|