module rename updates after rebase for unit tests

Author: paulb777

FirebaseAI/Tests/Unit/TemplateChatSessionTests.swift

@@ -12,7 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-@testable import FirebaseAI
+@testable import FirebaseAILogic
 import FirebaseCore
 import XCTest
 

FirebaseAI/Tests/Unit/TemplateGenerativeModelTests.swift

@@ -12,7 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-@testable import FirebaseAI
+@testable import FirebaseAILogic
 import FirebaseCore
 import XCTest
 

FirebaseAI/Tests/Unit/TemplateImagenModelTests.swift

@@ -12,7 +12,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-@testable import FirebaseAI
+@testable import FirebaseAILogic
 import XCTest
 
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)

FirebaseAI/Tests/Unit/TemplateVariableTests.swift

@@ -0,0 +1,29 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+@testable import FirebaseAILogic
+import XCTest
+
+@available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
+final class TemplateVariableTests: XCTestCase {
+  func testInitWithFloat() throws {
+    let floatValue: Float = 3.14
+    let templateVariable = try TemplateVariable(value: floatValue)
+    guard case let .double(doubleValue) = templateVariable else {
+      XCTFail("Expected a .double case, but got \(templateVariable)")
+      return
+    }
+    XCTAssertEqual(doubleValue, Double(floatValue), accuracy: 1e-6)
+  }
+}