Skip to content

Commit

Permalink
blacked my tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Nick Schouten authored and Nick Schouten committed Nov 20, 2024
1 parent 20d7586 commit 630b373
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions papermill/tests/test_iorw.py
Original file line number Diff line number Diff line change
Expand Up @@ -329,8 +329,7 @@ def test_read_with_auth(self):
text = 'request test response'
auth = 'Basic dW5hbWU6cGFzc3dvcmQK'

with patch.dict(os.environ, clear=True) as env,\
patch('papermill.iorw.requests.get') as mock_get:
with patch.dict(os.environ, clear=True) as env, patch('papermill.iorw.requests.get') as mock_get:
env['PAPERMILL_HTTP_AUTH_HEADER'] = auth
mock_get.return_value = Mock(text=text)

Expand All @@ -346,8 +345,7 @@ def test_read_with_accept_header(self):
text = 'request test response'
accept_type = 'test accept type'

with patch.dict(os.environ, clear=True) as env,\
patch('papermill.iorw.requests.get') as mock_get:
with patch.dict(os.environ, clear=True) as env, patch('papermill.iorw.requests.get') as mock_get:
env['PAPERMILL_HTTP_ACCEPT_HEADER'] = accept_type
mock_get.return_value = Mock(text=text)

Expand Down Expand Up @@ -375,8 +373,7 @@ def test_write_with_auth(self):
buf = '{"papermill": true}'
auth = 'token'

with patch.dict(os.environ, clear=True) as env,\
patch('papermill.iorw.requests.put') as mock_put:
with patch.dict(os.environ, clear=True) as env, patch('papermill.iorw.requests.put') as mock_put:
env['PAPERMILL_HTTP_AUTH_HEADER'] = auth

HttpHandler.write(buf, path)
Expand Down

0 comments on commit 630b373

Please sign in to comment.