Fix the tokenizer tests mocking globally

Author: kylecarbs

src/utils/tokens/tokenStatsCalculator.test.ts

@@ -1,11 +1,4 @@
-import { describe, test, expect, jest } from "@jest/globals";
-
-// Mock the tokenizer module before importing tokenStatsCalculator
-jest.mock("@/utils/main/tokenizer", () => ({
-  getTokenizerForModel: jest.fn(),
-  countTokensForData: jest.fn(),
-  getToolDefinitionTokens: jest.fn(),
-}));
+import { describe, test, expect, mock } from "bun:test";
 
 import {
   createDisplayUsage,
@@ -20,6 +13,13 @@ import {
 import type { LanguageModelV2Usage } from "@ai-sdk/provider";
 import type { MuxMessage } from "@/types/message";
 
+// Mock the tokenizer module before importing tokenStatsCalculator
+mock.module("@/utils/main/tokenizer", () => ({
+  getTokenizerForModel: jest.fn(),
+  countTokensForData: jest.fn(),
+  getToolDefinitionTokens: jest.fn(),
+}));
+
 describe("createDisplayUsage", () => {
   test("uses usage.reasoningTokens when available", () => {
     const usage: LanguageModelV2Usage = {