add swift test cases

This commit is contained in:
Te993 2025-01-10 13:55:59 +08:00
parent d3eeeae218
commit 61777707ca
2 changed files with 28 additions and 1 deletions

View file

@ -52,7 +52,7 @@ var omniVlmSources = [
]
var testSources = [
"tests/LlavaTests/LlavaTests.swift"
"swift/LlavaTests/LlavaTests.swift"
]
var resources: [Resource] = []

View file

@ -0,0 +1,27 @@
import XCTest
@testable import omnivlm
final class LlavaTests: XCTestCase {
func testOmniVlm() {
omnivlm_init("/Users/liute/Downloads/model-q4_0.gguf",
"/Users/liute/Downloads/projector-q4_0.gguf",
"vlm-81-instruct")
let startTime = Date()
if let cString = omnivlm_inference("describe the image", "/Users/liute/Downloads/cat.png") {
let res = String(cString: cString)
print("res: \(res)")
let endTime = Date()
let inferenceTime = endTime.timeIntervalSince(startTime)
print("Inference time: \(inferenceTime) seconds")
XCTAssertFalse(res.isEmpty, "res should not be null")
} else {
XCTFail("failed")
}
omnivlm_free()
}
}