fix(web): adjusts existing unit tests to match new intermediate-prediction-data format

Author: jahorton

web/src/test/auto/headless/engine/predictive-text/worker-thread/correction-search/early-correction-search-stopping.tests.ts

@@ -1,6 +1,16 @@
 import { assert } from 'chai';
 
-import { CORRECTION_SEARCH_THRESHOLDS, CorrectionPredictionTuple, ModelCompositor, shouldStopSearchingEarly } from "@keymanapp/lm-worker/test-index";
+import { CORRECTION_SEARCH_THRESHOLDS, IntermediateTokenizedPrediction, ModelCompositor, shouldStopSearchingEarly } from "@keymanapp/lm-worker/test-index";
+
+function mockTokenizedPrediction(value: number) {
+  return {
+    metadata: {
+      probabilities: {
+        total: value
+      }
+    }
+  } as IntermediateTokenizedPrediction
+}
 
 describe('correction-search: shouldStopSearchingEarly', () => {
   it('stops early once new corrections are less likely than currently discovered predictions', () => {
@@ -12,12 +22,7 @@ describe('correction-search: shouldStopSearchingEarly', () => {
     assert.equal(predictionProbs.length, ModelCompositor.MAX_SUGGESTIONS, "test setup no longer valid");
 
     // The only part for each entry we actually care about here:  .totalProb.
-    /** @type {import('#./predict-helpers.js').CorrectionPredictionTuple[]} */
-    const predictions = predictionProbs.map((entry) => {
-      return {
-        totalProb: entry
-      } as CorrectionPredictionTuple
-    });
+    const predictions = predictionProbs.map((entry) => mockTokenizedPrediction(entry));
 
     // Thresholding is performed in log-space.
     // 0.0501 and 0.0499 are offset on each side of 0.05, the last value in the array defined above.
@@ -33,8 +38,8 @@ describe('correction-search: shouldStopSearchingEarly', () => {
     //
     // Can technically run the method with an empty array, but the actual scenario would have
     // at least one prediction present in the "found predictions" array.
-    assert.isFalse(shouldStopSearchingEarly(baseCost, baseCost + expectedThreshold - 0.01, [{ totalProb: Math.exp(-1) } as CorrectionPredictionTuple]));
-    assert.isTrue(shouldStopSearchingEarly( baseCost, baseCost + expectedThreshold + 0.01, [{ totalProb: Math.exp(-1) } as CorrectionPredictionTuple]));
+    assert.isFalse(shouldStopSearchingEarly(baseCost, baseCost + expectedThreshold - 0.01, [mockTokenizedPrediction(Math.exp(-1))]));
+    assert.isTrue(shouldStopSearchingEarly( baseCost, baseCost + expectedThreshold + 0.01, [mockTokenizedPrediction(Math.exp(-1))]));
   });
 
   it('stops checking corrections earlier when enough predictions have been found', () => {
@@ -43,11 +48,7 @@ describe('correction-search: shouldStopSearchingEarly', () => {
 
     // The only part for each entry we actually care about here:  .totalProb.
     /** @type {import('#./predict-helpers.js').CorrectionPredictionTuple[]} */
-    const predictions = predictionProbs.map((entry) => {
-      return {
-        totalProb: entry
-      } as CorrectionPredictionTuple
-    });
+    const predictions = predictionProbs.map((entry) => mockTokenizedPrediction(entry));
 
     const baseCost = 1;