[Platform] Add more PHPStan array type annotations to bridge tests

Author: OskarStark

src/platform/src/Bridge/HuggingFace/Tests/Output/ClassificationResultTest.php

@@ -108,6 +108,9 @@ public function testFromArrayPreservesOrder()
         $this->assertSame('third', $result->classifications[2]->label);
     }
 
+    /**
+     * @param array{label: string, score: float} $classification
+     */
     #[TestDox('fromArray handles various label formats')]
     #[TestWith([['label' => '', 'score' => 0.5]])]
     #[TestWith([['label' => 'UPPERCASE', 'score' => 0.5]])]

src/platform/src/Bridge/HuggingFace/Tests/Output/QuestionAnsweringResultTest.php

@@ -68,6 +68,9 @@ public function testFromArray()
         $this->assertSame(0.87, $result->score);
     }
 
+    /**
+     * @param array{answer: string, start: int, end: int, score: float} $data
+     */
     #[TestDox('fromArray handles various data formats')]
     #[TestWith([['answer' => 'Yes', 'start' => 0, 'end' => 3, 'score' => 1.0]])]
     #[TestWith([['answer' => 'No answer found', 'start' => -1, 'end' => -1, 'score' => 0.0]])]

src/platform/src/Bridge/HuggingFace/Tests/Output/SentenceSimilarityResultTest.php

@@ -40,6 +40,9 @@ public function testConstructionWithEmptyArray()
         $this->assertCount(0, $result->similarities);
     }
 
+    /**
+     * @param array<int, float> $similarities
+     */
     #[TestDox('Constructor accepts various similarity arrays')]
     #[TestWith([[0.5]])]
     #[TestWith([[0.0, 1.0]])]
@@ -70,6 +73,9 @@ public function testFromArray()
         $this->assertSame(0.23, $result->similarities[3]);
     }
 
+    /**
+     * @param array<int, float> $data
+     */
     #[TestDox('fromArray handles various data formats')]
     #[TestWith([[]])] // Empty array
     #[TestWith([[0.999]])] // Single high similarity
@@ -90,6 +96,9 @@ public function testFromArrayWithVariousData(array $data)
         }
     }
 
+    /**
+     * @param array<int, float> $similarities
+     */
     #[TestDox('Special float values are handled correctly')]
     #[TestWith([[0.0, 1.0, 0.5]])] // Boundary values
     #[TestWith([[-1.0, 2.0]])] // Outside normal range

src/platform/src/Bridge/HuggingFace/Tests/Output/TableQuestionAnsweringResultTest.php

@@ -47,6 +47,11 @@ public function testConstructionWithAllParameters()
         $this->assertSame($aggregator, $result->aggregator);
     }
 
+    /**
+     * @param array<int, array<int, int>> $coordinates
+     * @param array<int, int|string>      $cells
+     * @param array<int, string>          $aggregator
+     */
     #[TestDox('Constructor accepts various parameter combinations')]
     #[TestWith(['Yes', [], [], []])]
     #[TestWith(['No', [[0, 1]], ['A1'], ['COUNT']])]
@@ -94,6 +99,9 @@ public function testFromArrayWithAllFields()
         $this->assertSame(['SUM'], $result->aggregator);
     }
 
+    /**
+     * @param array<string, mixed> $data
+     */
     #[TestDox('fromArray handles optional fields with default values')]
     #[TestWith([['answer' => 'Test', 'coordinates' => [[0, 0], [1, 1]]]])]
     #[TestWith([['answer' => 'Test', 'cells' => ['A1', 'B1']]])]
@@ -130,6 +138,9 @@ public function testFromArrayWithVariousCellTypes()
         $this->assertSame(['NONE'], $result->aggregator);
     }
 
+    /**
+     * @param array<string, mixed> $data
+     */
     #[TestDox('fromArray handles various aggregator formats')]
     #[TestWith([['answer' => 'Test', 'aggregator' => []]])]
     #[TestWith([['answer' => 'Test', 'aggregator' => ['NONE']]])]

src/platform/src/Bridge/Mistral/Tests/TokenOutputProcessorTest.php

@@ -127,6 +127,9 @@ public function testItHandlesMissingUsageFields()
         $this->assertNull($tokenUsage->totalTokens);
     }
 
+    /**
+     * @param array<string, mixed> $data
+     */
     private function createRawResponse(array $data = []): RawHttpResult
     {
         $rawResponse = $this->createStub(ResponseInterface::class);

src/platform/src/Bridge/Ollama/Tests/Contract/AssistantMessageNormalizerTest.php

@@ -46,6 +46,9 @@ public function testGetSupportedTypes()
         $this->assertSame([AssistantMessage::class => true], $this->normalizer->getSupportedTypes(null));
     }
 
+    /**
+     * @param array<string, mixed> $expectedOutput
+     */
     #[DataProvider('normalizeDataProvider')]
     public function testNormalize(AssistantMessage $message, array $expectedOutput)
     {

src/platform/src/Bridge/Ollama/Tests/OllamaResultConverterTest.php

@@ -217,13 +217,19 @@ public function testConvertThinkingStreamingResponse()
         $this->assertArrayHasKey('total_duration', $chunks->current()->raw);
     }
 
+    /**
+     * @return iterable<array<string, mixed>>
+     */
     private function generateConvertStreamingStream(): iterable
     {
         yield ['model' => 'deepseek-r1:latest', 'created_at' => '2025-10-29T17:15:49.631700779Z', 'message' => ['role' => 'assistant', 'content' => 'Hello'], 'done' => false];
         yield ['model' => 'deepseek-r1:latest', 'created_at' => '2025-10-29T17:15:49.905924913Z', 'message' => ['role' => 'assistant', 'content' => ' world!'], 'done' => true,
             'done_reason' => 'stop', 'total_duration' => 100, 'load_duration' => 10, 'prompt_eval_count' => 42, 'prompt_eval_duration' => 30, 'eval_count' => 17, 'eval_duration' => 60];
     }
 
+    /**
+     * @return iterable<array<string, mixed>>
+     */
     private function generateConvertThinkingStreamingStream(): iterable
     {
         yield ['model' => 'deepseek-r1:latest', 'created_at' => '2025-10-29T17:15:49.631700779Z', 'message' => ['role' => 'assistant', 'content' => '', 'thinking' => 'Thinking'], 'done' => false];

src/platform/src/Bridge/OpenAi/Tests/Contract/DocumentNormalizerTest.php

@@ -41,6 +41,9 @@ public function testGetSupportedTypes()
         $this->assertSame($expected, $normalizer->getSupportedTypes(null));
     }
 
+    /**
+     * @param array<string, mixed> $expected
+     */
     #[DataProvider('normalizeDataProvider')]
     public function testNormalize(Document $document, array $expected)
     {
@@ -51,6 +54,9 @@ public function testNormalize(Document $document, array $expected)
         $this->assertEquals($expected, $normalized);
     }
 
+    /**
+     * @return iterable<array{0: Document, 1: array<string, mixed>}>
+     */
     public static function normalizeDataProvider(): iterable
     {
         yield 'document from file' => [