wehub-resource-sync
26 lines
1.0 KiB
Swift
26 lines
1.0 KiB
Swift
import Testing
|
|
@testable import CodexBarCore
|
|
|
|
struct AntigravityModelLabelTests {
|
|
@Test
|
|
func `humanizes raw model ids when label matches model id`() {
|
|
#expect(AntigravityStatusSnapshot.humanizedModelID("gemini-3-pro-preview") == "Gemini 3 Pro Preview")
|
|
#expect(AntigravityStatusSnapshot.humanizedModelID("gemini-2.5-flash") == "Gemini 2.5 Flash")
|
|
#expect(AntigravityStatusSnapshot.humanizedModelID("example-3-1-pro-low") == "Example 3.1 Pro Low")
|
|
#expect(AntigravityStatusSnapshot.humanizedModelID("gpt-api-oss") == "GPT API OSS")
|
|
#expect(AntigravityStatusSnapshot.humanizedModelID("").isEmpty)
|
|
}
|
|
|
|
@Test
|
|
func `preserves custom model labels`() {
|
|
let quota = AntigravityModelQuota(
|
|
label: "Custom enterprise label",
|
|
modelId: "gemini-3-pro-preview",
|
|
remainingFraction: 1,
|
|
resetTime: nil,
|
|
resetDescription: nil)
|
|
|
|
#expect(AntigravityStatusSnapshot.quotaDisplayLabel(quota) == "Custom enterprise label")
|
|
}
|
|
}
|