File size: 2,761 Bytes
64f7370 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | 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)
}
|