Add pytest.mark.parametrize to try both upload.pypi and test.pypi as …

…repository URLs

Signed-off-by: Aditya Saky <aditya@saky.in>

tests/test_upload.py

@@ -288,12 +288,16 @@ def none_upload(*args, **settings_kwargs):
     assert "/foo/bar.crt" == upload_settings.cacert
 
 
-def test_check_status_code_for_wrong_repo_url(make_settings, capsys):
+@pytest.mark.parametrize('repository_url', [
+    "https://upload.pypi.org/",
+    "https://test.pypi.org/"
+])
+def test_check_status_code_for_wrong_repo_url(repository_url, make_settings,
+        capsys):
     upload_settings = make_settings()
 
     # override defaults to use incorrect URL
-    upload_settings.repository_config['repository'] = \
-        "https://upload.pypi.org"
+    upload_settings.repository_config['repository'] = repository_url
 
     with pytest.raises(HTTPError):
         upload.upload(upload_settings, [