Formatting

Author: tim-u

tests/integration/dataproc_magics/test_magics.py

@@ -134,9 +134,7 @@ def test_dpip_magic_loads(ipython_shell):
 
 
 @mock.patch.object(DataprocSparkSession, "addArtifacts")
-def test_dpip_install_single_package(
-    mock_add_artifacts, ipython_shell, capsys
-):
+def test_dpip_install_single_package(mock_add_artifacts, ipython_shell, capsys):
     """Test installing a single package with %dpip."""
     ipython_shell.run_line_magic("dpip", "install pandas")
     mock_add_artifacts.assert_called_once_with("pandas", pypi=True)
@@ -150,9 +148,7 @@ def test_dpip_install_multiple_packages_with_flags(
     mock_add_artifacts, ipython_shell, capsys
 ):
     """Test installing multiple packages with flags like -U."""
-    ipython_shell.run_line_magic(
-        "dpip", "install -U numpy scikit-learn"
-    )
+    ipython_shell.run_line_magic("dpip", "install -U numpy scikit-learn")
     calls = [
         mock.call("numpy", pypi=True),
         mock.call("scikit-learn", pypi=True),
@@ -168,9 +164,7 @@ def test_dpip_no_install_command(ipython_shell, capsys):
     """Test usage message when 'install' is missing."""
     ipython_shell.run_line_magic("dpip", "pandas")
     captured = capsys.readouterr()
-    assert (
-        "Usage: %dpip install <package1> <package2> ..." in captured.out
-    )
+    assert "Usage: %dpip install <package1> <package2> ..." in captured.out
     assert "No packages specified." in captured.out
 
 
@@ -196,9 +190,7 @@ def test_dpip_no_session(mock_add_artifacts, ipython_shell, capsys):
     "addArtifacts",
     side_effect=Exception("Install failed"),
 )
-def test_dpip_install_failure(
-    mock_add_artifacts, ipython_shell, capsys
-):
+def test_dpip_install_failure(mock_add_artifacts, ipython_shell, capsys):
     """Test error message on installation failure."""
     ipython_shell.run_line_magic("dpip", "install bad-package")
     mock_add_artifacts.assert_called_once_with("bad-package", pypi=True)