Fix pre-commit: nbstripout ipynb, trailing newline on test file, ruff format

Author: Eve Kazarian

doc/code/executor/attack/split_payload_attack.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "889d45ef",
+   "id": "0",
    "metadata": {},
    "source": [
     "# Split-Payload Attack (Multi-Turn)\n",
@@ -30,7 +30,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "41206e71",
+   "id": "1",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -50,7 +50,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "3468b00a",
+   "id": "2",
    "metadata": {},
    "source": [
     "## Configure the attack\n",
@@ -64,7 +64,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3f6c874f",
+   "id": "3",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -91,7 +91,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d54e34f2",
+   "id": "4",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -111,7 +111,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "1ceab006",
+   "id": "5",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -120,7 +120,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "81a3f77a",
+   "id": "6",
    "metadata": {},
    "source": [
     "## Examine results\n",
@@ -135,7 +135,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0e8b1129",
+   "id": "7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -150,7 +150,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "950f71d9",
+   "id": "8",
    "metadata": {},
    "outputs": [],
    "source": [

tests/unit/executor/attack/multi_turn/test_split_payload.py

@@ -309,9 +309,7 @@ def test_harm_description_setter(self):
 class TestSplitPayloadAttackInitialization:
     """Tests for SplitPayloadAttack initialization."""
 
-    def test_init_with_minimal_parameters(
-        self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock
-    ):
+    def test_init_with_minimal_parameters(self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock):
         adversarial_config = AttackAdversarialConfig(target=mock_adversarial_chat)
         attack = SplitPayloadAttack(
             objective_target=mock_objective_target,
@@ -369,9 +367,7 @@ def test_init_with_invalid_max_regenerations(
                 max_regenerations=max_regenerations,
             )
 
-    def test_init_with_invalid_design_pattern(
-        self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock
-    ):
+    def test_init_with_invalid_design_pattern(self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock):
         adversarial_config = AttackAdversarialConfig(target=mock_adversarial_chat)
         with pytest.raises(ValueError, match="design_pattern must be one of"):
             SplitPayloadAttack(
@@ -400,8 +396,8 @@ def test_init_rejects_adversarial_chat_missing_capability(
     ):
         from pyrit.prompt_target.common.target_capabilities import CapabilityName
 
-        mock_adversarial_chat.configuration.includes.side_effect = (
-            lambda *, capability: capability != CapabilityName.SYSTEM_PROMPT
+        mock_adversarial_chat.configuration.includes.side_effect = lambda *, capability: (
+            capability != CapabilityName.SYSTEM_PROMPT
         )
         adversarial_config = AttackAdversarialConfig(target=mock_adversarial_chat)
         with pytest.raises(ValueError, match="SplitPayloadAttack"):
@@ -428,9 +424,7 @@ def test_get_attack_scoring_config(
         assert config.objective_scorer == mock_objective_scorer
         assert config.refusal_scorer == mock_refusal_scorer
 
-    def test_init_with_converter_config(
-        self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock
-    ):
+    def test_init_with_converter_config(self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock):
         from pyrit.prompt_converter import Base64Converter
         from pyrit.prompt_normalizer import PromptConverterConfiguration
 
@@ -452,9 +446,7 @@ def test_init_with_converter_config(
 class TestContextValidation:
     """Tests for context validation."""
 
-    def test_validate_context_empty_objective(
-        self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock
-    ):
+    def test_validate_context_empty_objective(self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock):
         attack = SplitPayloadTestHelper.create_attack(
             objective_target=mock_objective_target,
             adversarial_chat=mock_adversarial_chat,
@@ -540,26 +532,24 @@ async def test_setup_tracks_adversarial_conversation(
 class TestDecomposition:
     """Tests for the decomposition phase."""
 
-    def test_parse_decomposition_valid(
-        self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock
-    ):
+    def test_parse_decomposition_valid(self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock):
         attack = SplitPayloadTestHelper.create_attack(
             objective_target=mock_objective_target,
             adversarial_chat=mock_adversarial_chat,
             max_turns=3,
         )
-        response = json.dumps({
-            "turns": ["Q1?", "Q2?", "Q3?"],
-            "rationale": "test",
-            "harm_description": "test harm",
-        })
+        response = json.dumps(
+            {
+                "turns": ["Q1?", "Q2?", "Q3?"],
+                "rationale": "test",
+                "harm_description": "test harm",
+            }
+        )
         result = attack._parse_decomposition_response(response)
         assert len(result["turns"]) == 3
         assert result["harm_description"] == "test harm"
 
-    def test_parse_decomposition_invalid_json(
-        self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock
-    ):
+    def test_parse_decomposition_invalid_json(self, mock_objective_target: MagicMock, mock_adversarial_chat: MagicMock):
         attack = SplitPayloadTestHelper.create_attack(
             objective_target=mock_objective_target,
             adversarial_chat=mock_adversarial_chat,
@@ -811,9 +801,7 @@ async def test_regenerate_turn_returns_text(
             prompt_normalizer=mock_prompt_normalizer,
         )
         basic_context.session = ConversationSession()
-        mock_prompt_normalizer.send_prompt_async.return_value = create_prompt_response(
-            text="Alternative question?"
-        )
+        mock_prompt_normalizer.send_prompt_async.return_value = create_prompt_response(text="Alternative question?")
 
         result = await attack._regenerate_turn_async(
             context=basic_context,
@@ -857,9 +845,7 @@ async def test_regenerate_turn_strips_quotes(
             prompt_normalizer=mock_prompt_normalizer,
         )
         basic_context.session = ConversationSession()
-        mock_prompt_normalizer.send_prompt_async.return_value = create_prompt_response(
-            text='"Alternative question?"'
-        )
+        mock_prompt_normalizer.send_prompt_async.return_value = create_prompt_response(text='"Alternative question?"')
 
         result = await attack._regenerate_turn_async(
             context=basic_context,
@@ -898,9 +884,7 @@ async def test_perform_success_all_turns_pass_and_assembled_harmful(
         mock_prompt_normalizer.send_prompt_async.return_value = sample_response
 
         with (
-            patch.object(
-                attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition
-            ),
+            patch.object(attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition),
             patch.object(attack, "_check_refusal_async", new_callable=AsyncMock, return_value=no_refusal_score),
             patch.object(
                 attack,
@@ -946,13 +930,9 @@ async def test_perform_failure_assembled_not_harmful(
         decomposition = {"turns": ["Q1?", "Q2?"], "harm_description": ""}
 
         with (
-            patch.object(
-                attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition
-            ),
+            patch.object(attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition),
             patch.object(attack, "_check_refusal_async", new_callable=AsyncMock, return_value=no_refusal_score),
-            patch.object(
-                attack, "_send_prompt_to_target_async", new_callable=AsyncMock, return_value=sample_response
-            ),
+            patch.object(attack, "_send_prompt_to_target_async", new_callable=AsyncMock, return_value=sample_response),
             patch.object(
                 attack,
                 "_score_assembled_conversation_async",
@@ -989,16 +969,10 @@ async def test_perform_failure_turn_refused_after_max_regenerations(
 
         # Refusal on every attempt
         with (
-            patch.object(
-                attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition
-            ),
+            patch.object(attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition),
             patch.object(attack, "_check_refusal_async", new_callable=AsyncMock, return_value=refusal_score),
-            patch.object(
-                attack, "_send_prompt_to_target_async", new_callable=AsyncMock, return_value=sample_response
-            ),
-            patch.object(
-                attack, "_regenerate_turn_async", new_callable=AsyncMock, return_value="Regenerated Q1?"
-            ),
+            patch.object(attack, "_send_prompt_to_target_async", new_callable=AsyncMock, return_value=sample_response),
+            patch.object(attack, "_regenerate_turn_async", new_callable=AsyncMock, return_value="Regenerated Q1?"),
         ):
             attack._memory.duplicate_conversation_excluding_last_turn.return_value = "new-conv-id"
             result = await attack._perform_async(context=basic_context)
@@ -1033,18 +1007,10 @@ async def test_perform_with_regeneration_then_success(
         refusal_results = [refusal_score, no_refusal_score]
 
         with (
-            patch.object(
-                attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition
-            ),
-            patch.object(
-                attack, "_check_refusal_async", new_callable=AsyncMock, side_effect=refusal_results
-            ),
-            patch.object(
-                attack, "_send_prompt_to_target_async", new_callable=AsyncMock, return_value=sample_response
-            ),
-            patch.object(
-                attack, "_regenerate_turn_async", new_callable=AsyncMock, return_value="Regenerated Q1?"
-            ),
+            patch.object(attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition),
+            patch.object(attack, "_check_refusal_async", new_callable=AsyncMock, side_effect=refusal_results),
+            patch.object(attack, "_send_prompt_to_target_async", new_callable=AsyncMock, return_value=sample_response),
+            patch.object(attack, "_regenerate_turn_async", new_callable=AsyncMock, return_value="Regenerated Q1?"),
             patch.object(
                 attack,
                 "_score_assembled_conversation_async",
@@ -1078,13 +1044,9 @@ async def test_perform_sets_atomic_attack_identifier(
         decomposition = {"turns": ["Q1?"], "harm_description": ""}
 
         with (
-            patch.object(
-                attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition
-            ),
+            patch.object(attack, "_decompose_objective_async", new_callable=AsyncMock, return_value=decomposition),
             patch.object(attack, "_check_refusal_async", new_callable=AsyncMock, return_value=no_refusal_score),
-            patch.object(
-                attack, "_send_prompt_to_target_async", new_callable=AsyncMock, return_value=sample_response
-            ),
+            patch.object(attack, "_send_prompt_to_target_async", new_callable=AsyncMock, return_value=sample_response),
             patch.object(
                 attack,
                 "_score_assembled_conversation_async",
@@ -1278,4 +1240,4 @@ async def test_complete_successful_attack(
         assert result.outcome == AttackOutcome.SUCCESS
         assert result.executed_turns == 3
         assert result.all_turns_passed_safety is True
-        assert result.assembled_payload_harmful is True
+        assert result.assembled_payload_harmful is True