@@ -329,8 +329,7 @@ def test_read_with_auth(self):
329
329
text = 'request test response'
330
330
auth = 'Basic dW5hbWU6cGFzc3dvcmQK'
331
331
332
- with patch .dict (os .environ , clear = True ) as env ,\
333
- patch ('papermill.iorw.requests.get' ) as mock_get :
332
+ with patch .dict (os .environ , clear = True ) as env , patch ('papermill.iorw.requests.get' ) as mock_get :
334
333
env ['PAPERMILL_HTTP_AUTH_HEADER' ] = auth
335
334
mock_get .return_value = Mock (text = text )
336
335
@@ -346,8 +345,7 @@ def test_read_with_accept_header(self):
346
345
text = 'request test response'
347
346
accept_type = 'test accept type'
348
347
349
- with patch .dict (os .environ , clear = True ) as env ,\
350
- patch ('papermill.iorw.requests.get' ) as mock_get :
348
+ with patch .dict (os .environ , clear = True ) as env , patch ('papermill.iorw.requests.get' ) as mock_get :
351
349
env ['PAPERMILL_HTTP_ACCEPT_HEADER' ] = accept_type
352
350
mock_get .return_value = Mock (text = text )
353
351
@@ -375,8 +373,7 @@ def test_write_with_auth(self):
375
373
buf = '{"papermill": true}'
376
374
auth = 'token'
377
375
378
- with patch .dict (os .environ , clear = True ) as env ,\
379
- patch ('papermill.iorw.requests.put' ) as mock_put :
376
+ with patch .dict (os .environ , clear = True ) as env , patch ('papermill.iorw.requests.put' ) as mock_put :
380
377
env ['PAPERMILL_HTTP_AUTH_HEADER' ] = auth
381
378
382
379
HttpHandler .write (buf , path )
0 commit comments