tests + hotfix

pytube/streams.py

@@ -135,6 +135,9 @@ class Stream(object):
     def filesize(self):
         """File size of the media stream in bytes."""
         headers = request.get(self.url, headers=True)
+        # https://github.com/nficano/pytube/issues/160
+        if 'content-length' in headers:
+            return int(headers['content-length'])
         return int(headers['Content-Length'])
 
     @property

tests/test_cipher.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+import pytest
+
+from pytube import cipher
+from pytube.exceptions import RegexMatchError
+
+
+def test_map_functions():
+    with pytest.raises(RegexMatchError):
+        cipher.map_functions('asdf')

tests/test_exceptions.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+from pytube.exceptions import ExtractError
+
+
+def test_is_expected():
+    try:
+        raise ExtractError('ppfff', video_id='YLnZklYFe7E')
+    except ExtractError as e:
+        assert e.video_id == 'YLnZklYFe7E'