|
|
|
from pytube import Playlist |
|
|
|
short_test_pl = 'https://www.youtube.com/watch?v=' \ |
|
'm5q2GCsteQs&list=PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw' |
|
long_test_pl = 'https://www.youtube.com/watch?v=' \ |
|
'9CHDoAsX1yo&list=UUXuqSBlHAE6Xw-yeJA0Tunw' |
|
|
|
|
|
def test_construct(): |
|
ob = Playlist(short_test_pl) |
|
expected = 'https://www.youtube.com/' \ |
|
'playlist?list=' \ |
|
'PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw' |
|
|
|
assert ob.construct_playlist_url() == expected |
|
|
|
|
|
def test_populate(): |
|
ob = Playlist(short_test_pl) |
|
expected = [ |
|
'https://www.youtube.com/watch?v=m5q2GCsteQs', |
|
'https://www.youtube.com/watch?v=5YK63cXyJ2Q', |
|
'https://www.youtube.com/watch?v=Rzt4rUPFYD4', |
|
] |
|
|
|
ob.populate_video_urls() |
|
assert ob.video_urls == expected |
|
|
|
|
|
def test_link_parse(): |
|
ob = Playlist(short_test_pl) |
|
|
|
expected = [ |
|
'/watch?v=m5q2GCsteQs', |
|
'/watch?v=5YK63cXyJ2Q', |
|
'/watch?v=Rzt4rUPFYD4', |
|
] |
|
assert ob.parse_links() == expected |
|
|
|
|
|
def test_download(): |
|
ob = Playlist(short_test_pl) |
|
ob.download_all() |
|
ob.download_all('.') |
|
ob.download_all(prefix_number=False) |
|
ob.download_all('.', prefix_number=False) |
|
|
|
|
|
def test_numbering(): |
|
ob = Playlist(short_test_pl) |
|
ob.populate_video_urls() |
|
gen = ob._path_num_prefix_generator(reverse=False) |
|
assert '1' in next(gen) |
|
assert '2' in next(gen) |
|
|
|
ob = Playlist(short_test_pl) |
|
ob.populate_video_urls() |
|
gen = ob._path_num_prefix_generator(reverse=True) |
|
assert str(len(ob.video_urls)) in next(gen) |
|
assert str(len(ob.video_urls) - 1) in next(gen) |
|
|
|
ob = Playlist(long_test_pl) |
|
ob.populate_video_urls() |
|
gen = ob._path_num_prefix_generator(reverse=False) |
|
nxt = next(gen) |
|
assert len(nxt) > 1 |
|
assert '1' in nxt |
|
nxt = next(gen) |
|
assert len(nxt) > 1 |
|
assert '2' in nxt |
|
|
|
ob = Playlist(long_test_pl) |
|
ob.populate_video_urls() |
|
gen = ob._path_num_prefix_generator(reverse=True) |
|
assert str(len(ob.video_urls)) in next(gen) |
|
assert str(len(ob.video_urls) - 1) in next(gen) |
|
|
