mock added to requirements to support 2.7

README.md

@@ -1,5 +1,10 @@
 # YouTube Transcript/Subtitle API (including automatically generated subtitles)
 
+[![Build Status](https://travis-ci.org/jdepoix/youtube-transcript-api.svg)](https://travis-ci.org/jdepoix/youtube-transcript-api)
+[![MIT license](http://img.shields.io/badge/license-MIT-brightgreen.svg?style=flat)
+[![image](https://img.shields.io/pypi/v/youtube-transcript-api.svg)](https://pypi.org/project/youtube-transcript-api/)
+[![image](https://img.shields.io/pypi/pyversions/youtube-transcript-api.svg)](https://pypi.org/project/youtube-transcript-api/)
+
 This is an python API which allows you to get the transcripts/subtitles for a given YouTube video. It also works for automatically generated subtitles and it does not require a headless browser, like other selenium based solutions do!
 
 ## Install

requirements.txt

@@ -1,5 +1,6 @@
 requests
 
 # testing
+mock
 httpretty
 coverage

setup.py

@@ -43,6 +43,7 @@ setuptools.setup(
         'requests',
     ],
     tests_require=[
+        'mock',
         'httpretty',
         'coverage',
     ],

youtube_transcript_api/test/test_api.py

@@ -1,5 +1,5 @@
 from unittest import TestCase
-from unittest.mock import MagicMock
+from mock import MagicMock
 
 import os
 

youtube_transcript_api/test/test_cli.py

@@ -1,5 +1,5 @@
 from unittest import TestCase
-from unittest.mock import MagicMock
+from mock import MagicMock
 
 import json