|
|
|
from pytube import playlist |
|
|
|
|
|
def test_construct(): |
|
ob = playlist.Playlist( |
|
"https://www.youtube.com/watch?v=m5q2GCsteQs&list=" |
|
"PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw") |
|
expected = "https://www.youtube.com/" \ |
|
"playlist?list=" \ |
|
"PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw" |
|
|
|
assert ob.construct_playlist_url() == expected |
|
|
|
|
|
def test_link_parse(): |
|
ob = playlist.Playlist( |
|
"https://www.youtube.com/watch?v=m5q2GCsteQs&list=" |
|
"PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw") |
|
|
|
expected = ["/watch?v=m5q2GCsteQs", |
|
"/watch?v=5YK63cXyJ2Q", |
|
"/watch?v=Rzt4rUPFYD4"] |
|
assert ob.parse_links() == expected |
|
|
|
|
|
def test_populate(): |
|
ob = playlist.Playlist( |
|
"https://www.youtube.com/watch?v=m5q2GCsteQs&list=" |
|
"PL525f8ds9RvsXDl44X6Wwh9t3fCzFNApw") |
|
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_download(): |
|
ob = playlist.Playlist( |
|
"https://www.youtube.com/watch?v=lByG_AgKS9k&list=" |
|
"PL525f8ds9RvuerPZ3bZygmNiYw2sP4BDk") |
|
ob.download_all() |
|
|
