| import ASRKit |
| import CryptoKit |
| import Foundation |
| import SpeechCore |
|
|
| enum CheckError: Error, CustomStringConvertible { |
| case failed(String) |
|
|
| var description: String { |
| switch self { |
| case .failed(let message): return message |
| } |
| } |
| } |
|
|
| func check(_ condition: @autoclosure () throws -> Bool, _ message: String) throws { |
| if try !condition() { |
| throw CheckError.failed(message) |
| } |
| } |
|
|
| func makeBundle(schema: Int, corrupt: Bool = false) throws -> URL { |
| let dir = FileManager.default.temporaryDirectory |
| .appendingPathComponent("spktest-\(UUID().uuidString)") |
| try FileManager.default.createDirectory(at: dir, withIntermediateDirectories: true) |
| let payload = Data("hello speechkit".utf8) |
| try payload.write(to: dir.appendingPathComponent("a.bin")) |
| var sha = SHA256.hash(data: payload).map { String(format: "%02x", $0) }.joined() |
| if corrupt { sha = String(sha.reversed()) } |
| let manifest: [String: Any] = [ |
| "schemaVersion": schema, |
| "bundleVersion": "0.0.1", |
| "assets": ["a": ["file": "a.bin", "sha256": sha, "sizeBytes": payload.count]], |
| ] |
| let data = try JSONSerialization.data(withJSONObject: manifest) |
| try data.write(to: dir.appendingPathComponent("manifest.json")) |
| return dir |
| } |
|
|
| func expectSpeechError(_ label: String, _ body: () throws -> Void) throws { |
| do { |
| try body() |
| throw CheckError.failed("\(label): expected SpeechError") |
| } catch is SpeechError { |
| return |
| } |
| } |
|
|
| do { |
| let validURL = try makeBundle(schema: 2) |
| let bundle = try AssetBundle(at: validURL, verifyHashes: true) |
| try check(try bundle.url("a").lastPathComponent == "a.bin", "asset URL should resolve") |
| try expectSpeechError("wrong schema") { _ = try AssetBundle(at: makeBundle(schema: 99)) } |
| try expectSpeechError("corrupted asset") { |
| _ = try AssetBundle(at: makeBundle(schema: 2, corrupt: true), verifyHashes: true) |
| } |
| try expectSpeechError("missing asset") { _ = try bundle.url("nonexistent") } |
|
|
| try check(MelExtractor.bucket(forFrames: 100) == 500, "100 frames should use 500 bucket") |
| try check(MelExtractor.bucket(forFrames: 500) == 500, "500 frames should use 500 bucket") |
| try check(MelExtractor.bucket(forFrames: 501) == 1000, "501 frames should use 1000 bucket") |
| try check(MelExtractor.bucket(forFrames: 2999) == 3000, "2999 frames should use 3000 bucket") |
| try check(MelExtractor.bucket(forFrames: 9999) == 3000, "long audio should clamp to 3000 bucket") |
| try check(AudioTower.arkAudioTokenCount(sampleCount: 67_263) == 52, |
| "4.2s reference should produce 52 audio tokens") |
|
|
| print("dev-check PASS") |
| } catch { |
| fputs("dev-check FAIL: \(error)\n", stderr) |
| exit(1) |
| } |
|
|
