From fee92ee7488e9c5e35c608adffbb738e28e5e419 Mon Sep 17 00:00:00 2001
From: kshitijk4poor <82637225+kshitijk4poor@users.noreply.github.com>
Date: Thu, 24 Sep 2026 17:09:14 +0530
Subject: [PATCH] test(agent): trim continuation-assembly tests to 2
 invariants; fix mock import path

The picked intentional-repetition loop test imported tests.run_agent.test_run_agent,
which does not exist on main (helpers live in tests/agent/test_run_agent.py).
---
 tests/agent/test_partial_stream_finish_reason.py | 15 +--------------
 1 file changed, 1 insertion(+), 14 deletions(-)

diff --git a/tests/agent/test_partial_stream_finish_reason.py b/tests/agent/test_partial_stream_finish_reason.py
index 76489f01ba..a8d57c32d7 100644
--- a/tests/agent/test_partial_stream_finish_reason.py
+++ b/tests/agent/test_partial_stream_finish_reason.py
@@ -359,19 +359,6 @@ class TestMixedToolCallsOneDroppedOneComplete:
 # ── Length-continuation prompt branching ──────────────────────────────────
 
 class TestLengthContinuationAssembly:
-    def test_repeated_complete_tail_is_not_joined_twice(self):
-        conclusion = (
-            "**Conclusion**: the project remains on standby until the stable "
-            "release. Nothing changes."
-        )
-
-        joined = _join_truncated_parts([
-            (f"Investigation details.\n\n{conclusion}", True),
-            (conclusion, False),
-        ])
-
-        assert joined.count(conclusion) == 1
-
     def test_distinct_continuation_is_preserved(self):
         assert _join_truncated_parts([
             ("The first half ends here", True),
@@ -492,7 +479,7 @@ class TestConversationLoopPartialStreamContinuation:
         assert result["final_response"].count(repeated_tail) == 1
 
     def test_output_limit_continuation_preserves_intentional_repetition(self, loop_agent):
-        from tests.run_agent.test_run_agent import _mock_response
+        from tests.agent.test_run_agent import _mock_response
 
         repeated = "This intentionally repeated sentence is longer than thirty-two characters."
         first = _mock_response(