unit tests

pytube/extract.py

@@ -20,7 +20,7 @@ def video_id(url):
 
 def watch_url(video_id):
     return (
-        'http://www.youtube.com/watch?v={video_id}'
+        'https://www.youtube.com/watch?v={video_id}'
         .format(video_id=video_id)
     )
 

tests/requirements.txt

@@ -5,3 +5,4 @@ mock==1.3.0
 pre-commit==0.15.0
 pytest==3.1.3
 pytest-cov==2.4.0
+pytest-sugar==0.9.0

tests/test_extract.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+"""
+tests.test_extract
+~~~~~~~~~~~~~~~~~~
+
+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://www.youtube.com/watch?v=9bZkp7q19f0'