hbmartin commited on
Commit
e29a6df
·
1 Parent(s): 0f90160

added test_display_progress_bar

Browse files
.idea/dictionaries/haroldmartin.xml CHANGED
@@ -4,6 +4,7 @@
4
  <w>acodec</w>
5
  <w>akamaized</w>
6
  <w>bitrate</w>
 
7
  <w>descr</w>
8
  <w>descramble</w>
9
  <w>descrambler</w>
 
4
  <w>acodec</w>
5
  <w>akamaized</w>
6
  <w>bitrate</w>
7
+ <w>capsys</w>
8
  <w>descr</w>
9
  <w>descramble</w>
10
  <w>descrambler</w>
tests/test_cli.py CHANGED
@@ -84,6 +84,12 @@ def test_download_caption_with_language_not_found(youtube):
84
  wrapped_all.assert_called()
85
 
86
 
 
 
 
 
 
 
87
  @mock.patch("pytube.Stream")
88
  @mock.patch("io.BufferedWriter")
89
  def test_on_progress(stream, writer):
 
84
  wrapped_all.assert_called()
85
 
86
 
87
+ def test_display_progress_bar(capsys):
88
+ cli.display_progress_bar(bytes_received=25, filesize=100, scale=0.55)
89
+ out, err = capsys.readouterr()
90
+ assert "25.0%" in out
91
+
92
+
93
  @mock.patch("pytube.Stream")
94
  @mock.patch("io.BufferedWriter")
95
  def test_on_progress(stream, writer):