Hugging Face's logo

Ferdowsi
/
pytube

Model card Files Community
hbmartin commited on
Commit
1e87762
·
1 Parent(s): e25f646

fix tests

Browse files
Files changed (1) hide show
  1. tests/test_streams.py +8 -8
tests/test_streams.py CHANGED
@@ -10,7 +10,7 @@ from pytube import Stream, streams
10
 
11
 
12
  def test_filesize(cipher_signature, mocker):
13
- mocker.patch.object(request, "headers")
14
  request.head.return_value = {"content-length": "6796391"}
15
  assert cipher_signature.streams.first().filesize == 6796391
16
 
@@ -87,7 +87,7 @@ def test_views(cipher_signature):
87
 
88
 
89
  def test_download(cipher_signature, mocker):
90
- mocker.patch.object(request, "headers")
91
  request.head.return_value = {"content-length": "16384"}
92
  mocker.patch.object(request, "stream")
93
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
@@ -97,7 +97,7 @@ def test_download(cipher_signature, mocker):
97
 
98
 
99
  def test_download_with_prefix(cipher_signature, mocker):
100
- mocker.patch.object(request, "headers")
101
  request.head.return_value = {"content-length": "16384"}
102
  mocker.patch.object(request, "stream")
103
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
@@ -109,7 +109,7 @@ def test_download_with_prefix(cipher_signature, mocker):
109
 
110
 
111
  def test_download_with_filename(cipher_signature, mocker):
112
- mocker.patch.object(request, "headers")
113
  request.head.return_value = {"content-length": "16384"}
114
  mocker.patch.object(request, "stream")
115
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
@@ -121,7 +121,7 @@ def test_download_with_filename(cipher_signature, mocker):
121
 
122
 
123
  def test_download_with_existing(cipher_signature, mocker):
124
- mocker.patch.object(request, "headers")
125
  request.head.return_value = {"content-length": "16384"}
126
  mocker.patch.object(request, "stream")
127
  streams.target_directory = MagicMock(return_value="/target")
@@ -137,7 +137,7 @@ def test_download_with_existing(cipher_signature, mocker):
137
 
138
 
139
  def test_download_with_existing_no_skip(cipher_signature, mocker):
140
- mocker.patch.object(request, "headers")
141
  request.head.return_value = {"content-length": "16384"}
142
  mocker.patch.object(request, "stream")
143
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
@@ -167,7 +167,7 @@ def test_on_progress_hook(cipher_signature, mocker):
167
  callback_fn = mock.MagicMock()
168
  cipher_signature.register_on_progress_callback(callback_fn)
169
 
170
- mocker.patch.object(request, "headers")
171
  request.head.return_value = {"content-length": "16384"}
172
  mocker.patch.object(request, "stream")
173
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
@@ -186,7 +186,7 @@ def test_on_complete_hook(cipher_signature, mocker):
186
  callback_fn = mock.MagicMock()
187
  cipher_signature.register_on_complete_callback(callback_fn)
188
 
189
- mocker.patch.object(request, "headers")
190
  request.head.return_value = {"content-length": "16384"}
191
  mocker.patch.object(request, "stream")
192
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
 
10
 
11
 
12
  def test_filesize(cipher_signature, mocker):
13
+ mocker.patch.object(request, "head")
14
  request.head.return_value = {"content-length": "6796391"}
15
  assert cipher_signature.streams.first().filesize == 6796391
16
 
 
87
 
88
 
89
  def test_download(cipher_signature, mocker):
90
+ mocker.patch.object(request, "head")
91
  request.head.return_value = {"content-length": "16384"}
92
  mocker.patch.object(request, "stream")
93
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
 
97
 
98
 
99
  def test_download_with_prefix(cipher_signature, mocker):
100
+ mocker.patch.object(request, "head")
101
  request.head.return_value = {"content-length": "16384"}
102
  mocker.patch.object(request, "stream")
103
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
 
109
 
110
 
111
  def test_download_with_filename(cipher_signature, mocker):
112
+ mocker.patch.object(request, "head")
113
  request.head.return_value = {"content-length": "16384"}
114
  mocker.patch.object(request, "stream")
115
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
 
121
 
122
 
123
  def test_download_with_existing(cipher_signature, mocker):
124
+ mocker.patch.object(request, "head")
125
  request.head.return_value = {"content-length": "16384"}
126
  mocker.patch.object(request, "stream")
127
  streams.target_directory = MagicMock(return_value="/target")
 
137
 
138
 
139
  def test_download_with_existing_no_skip(cipher_signature, mocker):
140
+ mocker.patch.object(request, "head")
141
  request.head.return_value = {"content-length": "16384"}
142
  mocker.patch.object(request, "stream")
143
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
 
167
  callback_fn = mock.MagicMock()
168
  cipher_signature.register_on_progress_callback(callback_fn)
169
 
170
+ mocker.patch.object(request, "head")
171
  request.head.return_value = {"content-length": "16384"}
172
  mocker.patch.object(request, "stream")
173
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])
 
186
  callback_fn = mock.MagicMock()
187
  cipher_signature.register_on_complete_callback(callback_fn)
188
 
189
+ mocker.patch.object(request, "head")
190
  request.head.return_value = {"content-length": "16384"}
191
  mocker.patch.object(request, "stream")
192
  request.stream.return_value = iter([str(random.getrandbits(8 * 1024))])