Merge pull request #10339 from nipunn1313/blacken_tests38

Author: uranusjr

tests/functional/test_search.py

@@ -13,8 +13,8 @@ def test_version_compare():
     Test version comparison.
 
     """
-    assert highest_version(['1.0', '2.0', '0.1']) == '2.0'
-    assert highest_version(['1.0a1', '1.0']) == '1.0'
+    assert highest_version(["1.0", "2.0", "0.1"]) == "2.0"
+    assert highest_version(["1.0a1", "1.0"]) == "1.0"
 
 
 def test_pypi_xml_transformation():
@@ -24,32 +24,32 @@ def test_pypi_xml_transformation():
     """
     pypi_hits = [
         {
-            'name': 'foo',
-            'summary': 'foo summary',
-            'version': '1.0',
+            "name": "foo",
+            "summary": "foo summary",
+            "version": "1.0",
         },
         {
-            'name': 'foo',
-            'summary': 'foo summary v2',
-            'version': '2.0',
+            "name": "foo",
+            "summary": "foo summary v2",
+            "version": "2.0",
         },
         {
-            '_pypi_ordering': 50,
-            'name': 'bar',
-            'summary': 'bar summary',
-            'version': '1.0',
+            "_pypi_ordering": 50,
+            "name": "bar",
+            "summary": "bar summary",
+            "version": "1.0",
         },
     ]
     expected = [
         {
-            'versions': ['1.0', '2.0'],
-            'name': 'foo',
-            'summary': 'foo summary v2',
+            "versions": ["1.0", "2.0"],
+            "name": "foo",
+            "summary": "foo summary v2",
         },
         {
-            'versions': ['1.0'],
-            'name': 'bar',
-            'summary': 'bar summary',
+            "versions": ["1.0"],
+            "name": "bar",
+            "summary": "bar summary",
         },
     ]
     assert transform_hits(pypi_hits) == expected
@@ -62,41 +62,41 @@ def test_basic_search(script):
     End to end test of search command.
 
     """
-    output = script.pip('search', 'pip')
+    output = script.pip("search", "pip")
     assert (
-        'The PyPA recommended tool for installing '
-        'Python packages.' in output.stdout
+        "The PyPA recommended tool for installing Python packages." in output.stdout
     )
 
 
 @pytest.mark.network
 @pytest.mark.skip(
-    reason=("Warehouse search behavior is different and no longer returns "
-            "multiple results. See "
-            "https://github.com/pypa/warehouse/issues/3717 for more "
-            "information."),
+    reason=(
+        "Warehouse search behavior is different and no longer returns "
+        "multiple results. See "
+        "https://github.com/pypa/warehouse/issues/3717 for more "
+        "information."
+    ),
 )
 @pytest.mark.search
 def test_multiple_search(script):
     """
     Test searching for multiple packages at once.
 
     """
-    output = script.pip('search', 'pip', 'INITools')
+    output = script.pip("search", "pip", "INITools")
     assert (
-        'The PyPA recommended tool for installing '
-        'Python packages.' in output.stdout
+        "The PyPA recommended tool for installing Python packages." in output.stdout
     )
-    assert 'Tools for parsing and using INI-style files' in output.stdout
+    assert "Tools for parsing and using INI-style files" in output.stdout
 
 
 @pytest.mark.search
 def test_search_missing_argument(script):
     """
     Test missing required argument for search
     """
-    result = script.pip('search', expect_error=True)
-    assert 'ERROR: Missing required argument (search query).' in result.stderr
+    result = script.pip("search", expect_error=True)
+    assert "ERROR: Missing required argument (search query)." in result.stderr
 
 
 @pytest.mark.network
@@ -105,7 +105,7 @@ def test_run_method_should_return_success_when_find_packages():
     """
     Test SearchCommand.run for found package
     """
-    command = create_command('search')
+    command = create_command("search")
     cmdline = "--index=https://pypi.org/pypi pip"
     with command.main_context():
         options, args = command.parse_args(cmdline.split())
@@ -119,7 +119,7 @@ def test_run_method_should_return_no_matches_found_when_does_not_find_pkgs():
     """
     Test SearchCommand.run for no matches
     """
-    command = create_command('search')
+    command = create_command("search")
     cmdline = "--index=https://pypi.org/pypi nonexistentpackage"
     with command.main_context():
         options, args = command.parse_args(cmdline.split())
@@ -133,7 +133,7 @@ def test_search_should_exit_status_code_zero_when_find_packages(script):
     """
     Test search exit status code for package found
     """
-    result = script.pip('search', 'pip')
+    result = script.pip("search", "pip")
     assert result.returncode == SUCCESS
 
 
@@ -143,7 +143,7 @@ def test_search_exit_status_code_when_finds_no_package(script):
     """
     Test search exit status code for no matches
     """
-    result = script.pip('search', 'nonexistentpackage', expect_error=True)
+    result = script.pip("search", "nonexistentpackage", expect_error=True)
     assert result.returncode == NO_MATCHES_FOUND, result.returncode
 
 
@@ -154,26 +154,24 @@ def test_latest_prerelease_install_message(caplog, monkeypatch):
     """
     hits = [
         {
-            'name': 'ni',
-            'summary': 'For knights who say Ni!',
-            'versions': ['1.0.0', '1.0.1a']
+            "name": "ni",
+            "summary": "For knights who say Ni!",
+            "versions": ["1.0.0", "1.0.1a"],
         }
     ]
 
     installed_package = pretend.stub(project_name="ni")
-    monkeypatch.setattr("pip._vendor.pkg_resources.working_set",
-                        [installed_package])
+    monkeypatch.setattr("pip._vendor.pkg_resources.working_set", [installed_package])
 
     dist = pretend.stub(version="1.0.0")
     get_dist = pretend.call_recorder(lambda x: dist)
-    monkeypatch.setattr("pip._internal.commands.search.get_distribution",
-                        get_dist)
+    monkeypatch.setattr("pip._internal.commands.search.get_distribution", get_dist)
     with caplog.at_level(logging.INFO):
         print_results(hits)
 
     message = caplog.records[-1].getMessage()
     assert 'pre-release; install with "pip install --pre"' in message
-    assert get_dist.calls == [pretend.call('ni')]
+    assert get_dist.calls == [pretend.call("ni")]
 
 
 @pytest.mark.search
@@ -183,20 +181,20 @@ def test_search_print_results_should_contain_latest_versions(caplog):
     """
     hits = [
         {
-            'name': 'testlib1',
-            'summary': 'Test library 1.',
-            'versions': ['1.0.5', '1.0.3']
+            "name": "testlib1",
+            "summary": "Test library 1.",
+            "versions": ["1.0.5", "1.0.3"],
         },
         {
-            'name': 'testlib2',
-            'summary': 'Test library 1.',
-            'versions': ['2.0.1', '2.0.3']
-        }
+            "name": "testlib2",
+            "summary": "Test library 1.",
+            "versions": ["2.0.1", "2.0.3"],
+        },
     ]
 
     with caplog.at_level(logging.INFO):
         print_results(hits)
 
     log_messages = sorted([r.getMessage() for r in caplog.records])
-    assert log_messages[0].startswith('testlib1 (1.0.5)')
-    assert log_messages[1].startswith('testlib2 (2.0.3)')
+    assert log_messages[0].startswith("testlib1 (1.0.5)")
+    assert log_messages[1].startswith("testlib2 (2.0.3)")