more linter

datarobot · Dec 17, 2018 · f903934 · f903934
1 parent d8cb066
commit f903934
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -252,7 +252,8 @@ def test_auto_small_dataset():
 def test_acquire_api_token(live_server):
     ui = PickableMock()
     base_url = '{webhost}/predApi/v1.0/'.format(webhost=live_server.url())
-    ret = acquire_api_token(base_url, {}, 'username', 'password', False, ui, False)
+    ret = acquire_api_token(base_url, {}, 'username', 'password', False, ui,
+                            False)
     assert ret == 'Som3tok3n'
     ui.info.assert_called_with('api-token acquired')
     ui.debug.assert_called_with('api-token: Som3tok3n')
@@ -262,7 +263,8 @@ def test_acquire_api_token_unauthorized(live_server):
     ui = PickableMock()
     base_url = '{webhost}/predApi/v1.0/'.format(webhost=live_server.url())
     with pytest.raises(ValueError) as ctx:
-        acquire_api_token(base_url, {}, 'unknown', 'passwd', False, ui, False)
+        acquire_api_token(base_url, {}, 'unknown', 'passwd', False, ui,
+                          False)
         assert not ui.info.called
         assert not ui.debug.called
     assert str(ctx.value) == 'wrong credentials'
@@ -272,7 +274,8 @@ def test_acquire_api_token_bad_status(live_server):
     ui = PickableMock()
     base_url = '{webhost}/predApi/v1.0/'.format(webhost=live_server.url())
     with pytest.raises(ValueError) as ctx:
-        acquire_api_token(base_url, {}, 'bad_status', 'passwd', False, ui, False)
+        acquire_api_token(base_url, {}, 'bad_status', 'passwd', False, ui,
+                          False)
         assert not ui.info.called
         assert not ui.debug.called
     assert str(ctx.value) == 'api_token request returned status code 500'
@@ -282,7 +285,8 @@ def test_acquire_api_token_no_token1(live_server):
     ui = PickableMock()
     base_url = '{webhost}/predApi/v1.0/'.format(webhost=live_server.url())
     with pytest.raises(ValueError) as ctx:
-        acquire_api_token(base_url, {}, 'no_token1', 'passwd', False, ui, False)
+        acquire_api_token(base_url, {}, 'no_token1', 'passwd', False, ui,
+                          False)
         assert not ui.info.called
         assert not ui.debug.called
     assert str(ctx.value) == ('no api-token registered; '
@@ -293,7 +297,8 @@ def test_acquire_api_token_no_token2(live_server):
     ui = PickableMock()
     base_url = '{webhost}/predApi/v1.0/'.format(webhost=live_server.url())
     with pytest.raises(ValueError) as ctx:
-        acquire_api_token(base_url, {}, 'no_token2', 'passwd', False, ui, False)
+        acquire_api_token(base_url, {}, 'no_token2', 'passwd', False, ui,
+                          False)
         assert not ui.info.called
         assert not ui.debug.called
     assert str(ctx.value) == ('no api-token registered; '
@@ -303,7 +308,8 @@ def test_acquire_api_token_no_token2(live_server):
 def test_create_and_acquire_api_token(live_server):
     ui = PickableMock()
     base_url = '{webhost}/predApi/v1.0/'.format(webhost=live_server.url())
-    ret = acquire_api_token(base_url, {}, 'username', 'password', True, ui, False)
+    ret = acquire_api_token(base_url, {}, 'username', 'password', True, ui,
+                            False)
     assert ret == 'Som3tok3n'
     ui.info.assert_called_with('api-token acquired')
     ui.debug.assert_called_with('api-token: Som3tok3n')