made all tests deterministic.

tests/test_pytube.py

@@ -2,21 +2,70 @@
 # -*- coding: utf-8 -*-
 
 import mock
-from nose.tools import eq_
+from nose.tools import eq_, raises
 from pytube import api
+from pytube.exceptions import MultipleObjectsReturned
 
 
 class TestPytube(object):
     def setUp(self):
-        url = 'http://www.youtube.com/watch?v=Ik-RsDGPI5Y'
-        mock_js = open('tests/mock_video.js')
-        mock_html = open('tests/mock_video.html')
+        url = 'http://www.youtube.com/watch?v=9bZkp7q19f0'
+        self.mock_js = open('tests/mock_video.js').read()
+        self.mock_html = open('tests/mock_video.html').read()
         with mock.patch('pytube.api.urlopen') as urlopen:
-            urlopen.return_value.read.return_value = mock_html.read()
+            urlopen.return_value.read.return_value = self.mock_html
             self.yt = api.YouTube()
-            self.yt._js_code = mock_js.read()
+            self.yt._js_code = self.mock_js
             self.yt.from_url(url)
 
     def test_get_video_id(self):
         """Resolve the video id from url"""
-        eq_(self.yt.video_id, 'Ik-RsDGPI5Y')
+        eq_(self.yt.video_id, '9bZkp7q19f0')
+
+    def test_auto_filename(self):
+        """Generate safe filename based on video title"""
+        expected = u'PSY - GANGNAM STYLE(\uac15\ub0a8\uc2a4\ud0c0\uc77c) MV'
+
+        eq_(self.yt.filename, expected)
+
+    def test_manual_filename(self):
+        """Manually set a filename"""
+        expected = u'PSY - Gangnam Style'
+
+        self.yt.set_filename(expected)
+        eq_(self.yt.filename, expected)
+
+    def test_get_all_videos(self):
+        """Get all videos"""
+        eq_(len(self.yt.get_videos()), 6)
+
+    def test_filter_video_by_extension(self):
+        """Filter videos by filetype"""
+        eq_(len(self.yt.filter('mp4')), 2)
+        eq_(len(self.yt.filter('3gp')), 2)
+        eq_(len(self.yt.filter('webm')), 1)
+        eq_(len(self.yt.filter('flv')), 1)
+
+    def test_filter_video_by_extension_and_resolution(self):
+        """Filter videos by file extension and resolution"""
+        eq_(len(self.yt.filter('mp4', '720p')), 1)
+        eq_(len(self.yt.filter('mp4', '1080p')), 0)
+
+    def test_filter_video_by_extension_resolution_profile(self):
+        """Filter videos by file extension, resolution, and profile"""
+        eq_(len(self.yt.filter('mp4', '360p', 'Baseline')), 1)
+
+    def test_filter_video_by_profile(self):
+        """Filter videos by file profile"""
+        eq_(len(self.yt.filter(profile='Simple')), 2)
+
+    def test_filter_video_by_resolution(self):
+        """Filter videos by resolution"""
+        eq_(len(self.yt.filter(resolution='360p')), 2)
+
+    @raises(MultipleObjectsReturned)
+    def test_get_multiple_itmes(self):
+        """.get(...) cannot return more than one video"""
+        self.yt.get(profile='Simple')
+        self.yt.get('mp4')
+        self.yt.get(resolution='240p')