Rocky
dcb66a9849
End Live Activities and release the audio session synchronously on applicationWillTerminate, and continue macOS local-model install, onboarding, and settings polish on this branch.
48 lines
1.5 KiB
Swift
48 lines
1.5 KiB
Swift
// LocalASRDownloadSourceSorterTests.swift
|
|
// OSGKeyboardTests
|
|
|
|
import XCTest
|
|
@testable import OSGKeyboardShared
|
|
|
|
final class LocalASRDownloadSourceSorterTests: XCTestCase {
|
|
|
|
private func source(_ type: String, priority: Int = 1) -> LocalASRDownloadSource {
|
|
LocalASRDownloadSource(
|
|
type: type,
|
|
priority: priority,
|
|
url: "https://example.com/\(type).tar.bz2",
|
|
baseURL: nil,
|
|
files: nil
|
|
)
|
|
}
|
|
|
|
func testChinaMainlandPrefersModelScope() {
|
|
let sources = [
|
|
source("github"),
|
|
source("huggingface"),
|
|
source("modelscope"),
|
|
]
|
|
let sorted = LocalASRDownloadSourceSorter.sorted(sources, region: Locale.Region("CN"))
|
|
XCTAssertEqual(sorted.map(\.type), ["modelscope", "huggingface", "github"])
|
|
}
|
|
|
|
func testGlobalPrefersHuggingFace() {
|
|
let sources = [
|
|
source("github"),
|
|
source("huggingface"),
|
|
source("modelscope"),
|
|
]
|
|
let sorted = LocalASRDownloadSourceSorter.sorted(sources, region: Locale.Region("US"))
|
|
XCTAssertEqual(sorted.map(\.type), ["huggingface", "github", "modelscope"])
|
|
}
|
|
|
|
func testSameTypeUsesPriority() {
|
|
let sources = [
|
|
source("github", priority: 2),
|
|
source("github", priority: 1),
|
|
]
|
|
let sorted = LocalASRDownloadSourceSorter.sorted(sources, region: Locale.Region("US"))
|
|
XCTAssertEqual(sorted.map(\.priority), [1, 2])
|
|
}
|
|
}
|