style: format test files for CI compatibility

Run ruff format on 8 test files missed in previous commit.
CI checks: contribai/ AND tests/ directories.

Author: tang-vu

tests/integration/test_pipeline.py

@@ -87,32 +87,39 @@ async def test_dry_run_skips_pr_creation(
         pipeline._discovery.discover = AsyncMock(return_value=[mock_repo])
 
         pipeline._analyzer = AsyncMock()
-        pipeline._analyzer.analyze = AsyncMock(return_value=AnalysisResult(
-            repo=mock_repo,
-            findings=[sample_finding],
-            analyzed_files=1,
-            analysis_duration_sec=0.5,
-        ))
+        pipeline._analyzer.analyze = AsyncMock(
+            return_value=AnalysisResult(
+                repo=mock_repo,
+                findings=[sample_finding],
+                analyzed_files=1,
+                analysis_duration_sec=0.5,
+            )
+        )
 
         pipeline._generator = AsyncMock()
         pipeline._generator.generate = AsyncMock(return_value=sample_contribution)
 
         pipeline._pr_manager = AsyncMock()
-        pipeline._pr_manager.create_pr = AsyncMock(return_value=PRResult(
-            repo=mock_repo,
-            contribution=sample_contribution,
-            pr_number=42,
-            pr_url="https://github.com/test/pr/42",
-        ))
+        pipeline._pr_manager.create_pr = AsyncMock(
+            return_value=PRResult(
+                repo=mock_repo,
+                contribution=sample_contribution,
+                pr_number=42,
+                pr_url="https://github.com/test/pr/42",
+            )
+        )
 
         # Use real Memory with tmp_path
         from contribai.orchestrator.memory import Memory
+
         pipeline._memory = Memory(pipeline_config.storage.resolved_db_path)
         await pipeline._memory.init()
 
-        pipeline._github.get_file_tree = AsyncMock(return_value=[
-            FileNode(path="main.py", type="blob", size=500, sha="abc"),
-        ])
+        pipeline._github.get_file_tree = AsyncMock(
+            return_value=[
+                FileNode(path="main.py", type="blob", size=500, sha="abc"),
+            ]
+        )
         pipeline._github.get_file_content = AsyncMock(return_value="import unused\nprint('hello')")
 
         # Patch _init_components to be a no-op (components already set)
@@ -148,12 +155,15 @@ async def test_analyze_only_mode(self, pipeline_config, mock_repo):
         )
 
         pipeline._analyzer = AsyncMock()
-        pipeline._analyzer.analyze = AsyncMock(return_value=AnalysisResult(
-            repo=mock_repo,
-            findings=[finding],
-        ))
+        pipeline._analyzer.analyze = AsyncMock(
+            return_value=AnalysisResult(
+                repo=mock_repo,
+                findings=[finding],
+            )
+        )
 
         from contribai.orchestrator.memory import Memory
+
         pipeline._memory = Memory(pipeline_config.storage.resolved_db_path)
         await pipeline._memory.init()
 

tests/unit/test_analyzer.py

@@ -113,10 +113,20 @@ def test_removes_duplicates(self, analyzer):
 
     def test_keeps_different_findings(self, analyzer):
         findings = [
-            Finding(type=ContributionType.SECURITY_FIX, severity=Severity.HIGH,
-                    title="Issue A", description="", file_path="a.py"),
-            Finding(type=ContributionType.CODE_QUALITY, severity=Severity.MEDIUM,
-                    title="Issue B", description="", file_path="b.py"),
+            Finding(
+                type=ContributionType.SECURITY_FIX,
+                severity=Severity.HIGH,
+                title="Issue A",
+                description="",
+                file_path="a.py",
+            ),
+            Finding(
+                type=ContributionType.CODE_QUALITY,
+                severity=Severity.MEDIUM,
+                title="Issue B",
+                description="",
+                file_path="b.py",
+            ),
         ]
         result = analyzer._deduplicate(findings)
         assert len(result) == 2
@@ -125,12 +135,27 @@ def test_keeps_different_findings(self, analyzer):
 class TestFilterSeverity:
     def test_filters_below_threshold(self, analyzer):
         findings = [
-            Finding(type=ContributionType.SECURITY_FIX, severity=Severity.LOW,
-                    title="Low", description="", file_path="a.py"),
-            Finding(type=ContributionType.SECURITY_FIX, severity=Severity.MEDIUM,
-                    title="Medium", description="", file_path="b.py"),
-            Finding(type=ContributionType.SECURITY_FIX, severity=Severity.HIGH,
-                    title="High", description="", file_path="c.py"),
+            Finding(
+                type=ContributionType.SECURITY_FIX,
+                severity=Severity.LOW,
+                title="Low",
+                description="",
+                file_path="a.py",
+            ),
+            Finding(
+                type=ContributionType.SECURITY_FIX,
+                severity=Severity.MEDIUM,
+                title="Medium",
+                description="",
+                file_path="b.py",
+            ),
+            Finding(
+                type=ContributionType.SECURITY_FIX,
+                severity=Severity.HIGH,
+                title="High",
+                description="",
+                file_path="c.py",
+            ),
         ]
         result = analyzer._filter_severity(findings)
         assert len(result) == 2  # medium threshold: medium + high

tests/unit/test_generator.py

@@ -95,7 +95,7 @@ def test_docs_title(self, generator):
 
 class TestParseChanges:
     def test_parse_json_response(self, generator):
-        response = '''Here is the fix:
+        response = """Here is the fix:
 ```json
 {
   "changes": [
@@ -106,16 +106,16 @@ def test_parse_json_response(self, generator):
     }
   ]
 }
-```'''
+```"""
         changes = generator._parse_changes(response)
         assert len(changes) == 1
         assert changes[0].path == "src/config.py"
         assert not changes[0].is_new_file
 
     def test_parse_new_file(self, generator):
-        response = '''```json
+        response = """```json
 {"changes": [{"path": "new_file.py", "content": "print('hello')", "is_new_file": true}]}
-```'''
+```"""
         changes = generator._parse_changes(response)
         assert len(changes) == 1
         assert changes[0].is_new_file is True

tests/unit/test_github_client.py

@@ -70,7 +70,9 @@ def test_auth_header(self, client):
 class TestContributingGuide:
     @pytest.mark.asyncio
     async def test_returns_none_when_not_found(self, client):
-        client.get_file_content = AsyncMock(side_effect=GitHubAPIError("Not found", status_code=404))
+        client.get_file_content = AsyncMock(
+            side_effect=GitHubAPIError("Not found", status_code=404)
+        )
         result = await client.get_contributing_guide("owner", "repo")
         assert result is None
 

tests/unit/test_issue_solver.py

@@ -14,17 +14,26 @@ def solver(mock_llm, mock_github):
 
 @pytest.fixture
 def bug_issue():
-    return Issue(number=1, title="App crashes on login", body="TypeError when user logs in", labels=["bug"])
+    return Issue(
+        number=1, title="App crashes on login", body="TypeError when user logs in", labels=["bug"]
+    )
 
 
 @pytest.fixture
 def feature_issue():
-    return Issue(number=2, title="Add dark mode support", body="Please add dark theme toggle", labels=["enhancement"])
+    return Issue(
+        number=2,
+        title="Add dark mode support",
+        body="Please add dark theme toggle",
+        labels=["enhancement"],
+    )
 
 
 @pytest.fixture
 def docs_issue():
-    return Issue(number=3, title="Fix typo in README", body="Line 42 has a typo", labels=["documentation"])
+    return Issue(
+        number=3, title="Fix typo in README", body="Line 42 has a typo", labels=["documentation"]
+    )
 
 
 @pytest.fixture
@@ -76,7 +85,8 @@ def test_long_body_complex(self, solver):
 
     def test_many_file_references(self, solver):
         issue = Issue(
-            number=3, title="Fix stuff",
+            number=3,
+            title="Fix stuff",
             body="Change src/a.py src/b.py src/c.py src/d.py src/e.py",
             labels=[],
         )
@@ -91,7 +101,8 @@ def test_filters_basic(self, solver, bug_issue, docs_issue):
 
     def test_filters_complex(self, solver):
         complex_issue = Issue(
-            number=99, title="Redesign everything",
+            number=99,
+            title="Redesign everything",
             body="x" * 10000 + " file1.py file2.py file3.py file4.py",
             labels=[],
         )
@@ -102,11 +113,13 @@ def test_filters_complex(self, solver):
 class TestSolveIssue:
     @pytest.mark.asyncio
     async def test_solve_returns_finding(self, solver, bug_issue, sample_repo):
-        solver._llm.complete = AsyncMock(return_value="""FILE_PATH: src/auth.py
+        solver._llm.complete = AsyncMock(
+            return_value="""FILE_PATH: src/auth.py
 SEVERITY: high
 TITLE: Fix TypeError in login handler
 DESCRIPTION: The login handler raises TypeError when user object is None
-SUGGESTION: Add null check before accessing user properties""")
+SUGGESTION: Add null check before accessing user properties"""
+        )
 
         context = RepoContext(
             repo=sample_repo,

tests/unit/test_models.py

@@ -81,10 +81,20 @@ def test_top_findings_sorted(self, sample_repo):
 
     def test_filter_by_type(self, sample_repo):
         findings = [
-            Finding(type=ContributionType.SECURITY_FIX, severity=Severity.HIGH,
-                    title="Sec", description="", file_path="a.py"),
-            Finding(type=ContributionType.DOCS_IMPROVE, severity=Severity.LOW,
-                    title="Doc", description="", file_path="b.py"),
+            Finding(
+                type=ContributionType.SECURITY_FIX,
+                severity=Severity.HIGH,
+                title="Sec",
+                description="",
+                file_path="a.py",
+            ),
+            Finding(
+                type=ContributionType.DOCS_IMPROVE,
+                severity=Severity.LOW,
+                title="Doc",
+                description="",
+                file_path="b.py",
+            ),
         ]
         result = AnalysisResult(repo=sample_repo, findings=findings)
         sec = result.filter_by_type(ContributionType.SECURITY_FIX)
@@ -93,12 +103,27 @@ def test_filter_by_type(self, sample_repo):
 
     def test_filter_by_severity(self, sample_repo):
         findings = [
-            Finding(type=ContributionType.SECURITY_FIX, severity=Severity.LOW,
-                    title="Low", description="", file_path="a.py"),
-            Finding(type=ContributionType.SECURITY_FIX, severity=Severity.HIGH,
-                    title="High", description="", file_path="b.py"),
-            Finding(type=ContributionType.SECURITY_FIX, severity=Severity.CRITICAL,
-                    title="Crit", description="", file_path="c.py"),
+            Finding(
+                type=ContributionType.SECURITY_FIX,
+                severity=Severity.LOW,
+                title="Low",
+                description="",
+                file_path="a.py",
+            ),
+            Finding(
+                type=ContributionType.SECURITY_FIX,
+                severity=Severity.HIGH,
+                title="High",
+                description="",
+                file_path="b.py",
+            ),
+            Finding(
+                type=ContributionType.SECURITY_FIX,
+                severity=Severity.CRITICAL,
+                title="Crit",
+                description="",
+                file_path="c.py",
+            ),
         ]
         result = AnalysisResult(repo=sample_repo, findings=findings)
         high_plus = result.filter_by_severity(Severity.HIGH)

tests/unit/test_pr_manager.py

@@ -98,24 +98,24 @@ async def test_open_pr(self, pr_manager):
 
     @pytest.mark.asyncio
     async def test_merged_pr(self, pr_manager):
-        pr_manager._github._get = AsyncMock(
-            return_value={"state": "closed", "merged": True}
-        )
+        pr_manager._github._get = AsyncMock(return_value={"state": "closed", "merged": True})
         status = await pr_manager.get_pr_status("owner", "repo", 1)
         assert status == PRStatus.MERGED
 
     @pytest.mark.asyncio
     async def test_closed_pr(self, pr_manager):
-        pr_manager._github._get = AsyncMock(
-            return_value={"state": "closed", "merged": False}
-        )
+        pr_manager._github._get = AsyncMock(return_value={"state": "closed", "merged": False})
         status = await pr_manager.get_pr_status("owner", "repo", 1)
         assert status == PRStatus.CLOSED
 
     @pytest.mark.asyncio
     async def test_review_requested(self, pr_manager):
         pr_manager._github._get = AsyncMock(
-            return_value={"state": "open", "merged": False, "requested_reviewers": [{"login": "reviewer"}]}
+            return_value={
+                "state": "open",
+                "merged": False,
+                "requested_reviewers": [{"login": "reviewer"}],
+            }
         )
         status = await pr_manager.get_pr_status("owner", "repo", 1)
         assert status == PRStatus.REVIEW_REQUESTED

tests/unit/test_scorer.py

@@ -30,10 +30,12 @@ def good_contribution():
         contribution_type=ContributionType.SECURITY_FIX,
         title="Fix SQL injection in db.py",
         description="Parameterized all raw SQL queries to prevent SQL injection attacks.",
-        changes=[FileChange(
-            path="db.py",
-            new_content="""import sqlite3\n\ndef query(db, user_id):\n    cursor = db.execute('SELECT * FROM users WHERE id = ?', (user_id,))\n    return cursor.fetchall()\n""",
-        )],
+        changes=[
+            FileChange(
+                path="db.py",
+                new_content="""import sqlite3\n\ndef query(db, user_id):\n    cursor = db.execute('SELECT * FROM users WHERE id = ?', (user_id,))\n    return cursor.fetchall()\n""",
+            )
+        ],
         commit_message="fix(security): parameterize sql queries to prevent injection",
         branch_name="contribai/fix/sql-injection",
     )
@@ -69,10 +71,19 @@ def test_bad_contribution_fails(self, scorer, bad_contribution):
 
     def test_empty_changes_fail(self, scorer):
         contrib = Contribution(
-            finding=Finding(type=ContributionType.BUG_FIX if hasattr(ContributionType, "BUG_FIX") else ContributionType.CODE_QUALITY,
-                            severity=Severity.LOW, title="X", description="X", file_path="x.py"),
+            finding=Finding(
+                type=ContributionType.BUG_FIX
+                if hasattr(ContributionType, "BUG_FIX")
+                else ContributionType.CODE_QUALITY,
+                severity=Severity.LOW,
+                title="X",
+                description="X",
+                file_path="x.py",
+            ),
             contribution_type=ContributionType.CODE_QUALITY,
-            title="X", description="X", changes=[],
+            title="X",
+            description="X",
+            changes=[],
         )
         report = scorer.evaluate(contrib)
         assert report.checks["has_changes"].passed is False