Update tests/Builder/Stage/ScoreFusionStageTest.php

Co-authored-by: Jérôme Tamarelle <jerome@tamarelle.net>

Author: kraenhansen

tests/Builder/Stage/ScoreFusionStageTest.php

@@ -19,55 +19,48 @@ public function testExample(): void
             Stage::scoreFusion(
                 input: [
                     'pipelines' => [
-                        'searchOne' => [
-                            [
-                                '$vectorSearch' => [
-                                    'index' => 'vector_index',
-                                    'path' => 'plot_embedding',
-                                    'queryVector' => [-0.0016261312, -0.028070757, -0.011342932],
-                                    'numCandidates' => 150,
-                                    'limit' => 10,
-                                ],
-                            ],
-                        ],
-                        'searchTwo' => [
-                            [
-                                '$search' => [
-                                    'index' => '<INDEX_NAME>',
-                                    'text' => [
-                                        'query' => '<QUERY_TERM>',
-                                        'path' => '<FIELD_NAME>',
-                                    ],
-                                ],
-                            ],
-                        ],
+                        'searchOne' => new Pipeline(
+                            Stage::vectorSearch(
+                                index: 'vector_index',
+                                path: 'plot_embedding',
+                                queryVector: [-0.0016261312, -0.028070757, -0.011342932],
+                                numCandidates: 150,
+                                limit: 10,
+                            ),
+                        ),
+                        'searchTwo' => new Pipeline(
+                            Stage::search(
+                                Search::text(
+                                    query: '<QUERY_TERM>',
+                                    path: '<FIELD_NAME>',
+                                ),
+                                index: '<INDEX_NAME>',
+                            ),
+                        ),
                     ],
                     'normalization' => 'sigmoid',
                 ],
                 combination: [
                     'method' => 'expression',
-                    'expression' => [
-                        '$sum' => [
-                            [
-                                '$multiply' => [
-                                    '$$searchOne',
-                                    10,
-                                ],
-                            ],
-                            '$$searchTwo',
-                        ],
-                    ],
+                    'expression' => Expression::sum(
+                        Expression::multiply(
+                            Expression::variable('searchOne'),
+                            10,
+                        ),
+                        Expression::variable('searchTwo'),
+                    ),
                 ],
                 scoreDetails: true,
             ),
             Stage::project(
                 _id: 1,
                 title: 1,
                 plot: 1,
-                scoreDetails: ['$meta' => 'scoreDetails'],
+                scoreDetails: Expression::meta('scoreDetails'),
             ),
             Stage::limit(20),
         );
+
         $this->assertSamePipeline(Pipelines::ScoreFusionExample, $pipeline);
     }
 }