File size: 1,271 Bytes
bbf7f6e 5978020 bbf7f6e 5978020 42f45d6 bbf7f6e 42f45d6 bbf7f6e 42f45d6 bbf7f6e 42f45d6 5978020 bbf7f6e 42f45d6 5978020 bbf7f6e 42f45d6 5978020 bbf7f6e 42f45d6 5978020 bbf7f6e 42f45d6 5978020 bbf7f6e 5978020 42f45d6 5978020 bbf7f6e 42f45d6 5978020 bbf7f6e 42f45d6 5978020 |
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 |
#!/usr/bin/env/python
# -*- coding: utf-8 -*-
from nose.tools import eq_, raises
from pytube import YouTube
from pytube.exceptions import MultipleObjectsReturned
class TestYouTube(object):
"""Test all methods of Youtube class"""
def setUp(self):
"""Set up the all attributes required for a particular video."""
self.url = "https://www.youtube.com/watch?v=Ik-RsDGPI5Y"
self.video_id = 'Ik-RsDGPI5Y'
self.filename = 'Pulp Fiction - Dancing Scene'
self.yt = YouTube(self.url)
# using flv since it has only once video
self.flv = '<Video: Sorenson H.263 (.flv) - 240p - N/A>'
def test_url(self):
eq_(self.yt.url, self.url)
def test_video_id(self):
eq_(self.yt.video_id, self.video_id)
def test_filename(self):
eq_(self.yt.filename, self.filename)
def test_get_videos(self):
eq_(self.yt.get_videos())
def test_get_video_data(self):
eq_((self.yt.get_video_data()['args']['loaderUrl']), self.url)
@raises(MultipleObjectsReturned)
def test_get_false(self):
self.yt.get()
def test_get_true(self):
eq_(str(self.yt.get('flv')), self.flv)
def test_filter(self):
eq_(str(self.yt.filter('flv')[0]), self.flv)
|