nficano commited on
Commit
e83045b
·
1 Parent(s): 8b82476

itag test coverage

Browse files
Files changed (2) hide show
  1. tests/test_itags.py +12 -0
  2. tests/test_streams.py +0 -0
tests/test_itags.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ from pytube import itags
3
+
4
+
5
+ def test_get_format_profile():
6
+ profile = itags.get_format_profile(22)
7
+ assert profile['resolution'] == '720p'
8
+
9
+
10
+ def test_get_format_profile_non_existant():
11
+ profile = itags.get_format_profile(2239)
12
+ assert profile['resolution'] is None
tests/test_streams.py ADDED
File without changes