rocky
eaacc35cf7
Regeneration merges the on-device learned evidence with dictation and reply choices since the last generation instead of re-reading only the newest ~2,500 characters. It unlocks after 500 new characters, fades traits that stop getting support, and fully relearns styles without saved evidence.
104 lines
3.6 KiB
Swift
104 lines
3.6 KiB
Swift
// PersonalStyleLearningStateStoreTests.swift
|
|
// OSGKeyboard · Tests
|
|
//
|
|
// Verifies the device-local prior only applies to the exact saved generation.
|
|
|
|
@testable import OSGKeyboardShared
|
|
import XCTest
|
|
|
|
@MainActor
|
|
final class PersonalStyleLearningStateStoreTests: XCTestCase {
|
|
private var suiteName: String!
|
|
private var defaults: UserDefaults!
|
|
private var store: PersonalStyleLearningStateStore!
|
|
|
|
override func setUp() {
|
|
super.setUp()
|
|
suiteName = "com.osgkeyboard.personal-style-state.\(UUID().uuidString)"
|
|
defaults = UserDefaults(suiteName: suiteName)!
|
|
defaults.removePersistentDomain(forName: suiteName)
|
|
store = PersonalStyleLearningStateStore(defaults: defaults)
|
|
}
|
|
|
|
override func tearDown() {
|
|
defaults.removePersistentDomain(forName: suiteName)
|
|
super.tearDown()
|
|
}
|
|
|
|
func testSavedPriorMatchesSamePackAndGeneration() {
|
|
let pack = Self.pack(id: "user.style", generatedAt: Date(timeIntervalSince1970: 100))
|
|
store.save(Self.prior, for: pack)
|
|
|
|
XCTAssertEqual(store.prior(for: pack), Self.prior)
|
|
}
|
|
|
|
func testPriorSurvivesManualEditThatKeepsMetadata() {
|
|
let pack = Self.pack(id: "user.style", generatedAt: Date(timeIntervalSince1970: 100))
|
|
store.save(Self.prior, for: pack)
|
|
var edited = pack
|
|
edited.prompt = "# 角色\n手动修改\n# 风格边界\n自然\n# 示例\n输入 → 输出"
|
|
|
|
XCTAssertEqual(store.prior(for: edited), Self.prior)
|
|
}
|
|
|
|
func testPriorIsIgnoredForAnotherGenerationOrPack() {
|
|
store.save(
|
|
Self.prior,
|
|
for: Self.pack(id: "user.style", generatedAt: Date(timeIntervalSince1970: 100))
|
|
)
|
|
|
|
XCTAssertNil(
|
|
store.prior(
|
|
for: Self.pack(id: "user.style", generatedAt: Date(timeIntervalSince1970: 500))
|
|
)
|
|
)
|
|
XCTAssertNil(
|
|
store.prior(
|
|
for: Self.pack(id: "user.other", generatedAt: Date(timeIntervalSince1970: 100))
|
|
)
|
|
)
|
|
XCTAssertNil(store.prior(for: PolishStylePack(id: "user.style", name: "Legacy", prompt: "p")))
|
|
}
|
|
|
|
func testClearRemovesPrior() {
|
|
let pack = Self.pack(id: "user.style", generatedAt: Date(timeIntervalSince1970: 100))
|
|
store.save(Self.prior, for: pack)
|
|
store.clear()
|
|
|
|
XCTAssertNil(store.prior(for: pack))
|
|
}
|
|
|
|
private static let prior = PolishStyleLearningPriorEvidence(
|
|
asr: .init(
|
|
traits: [
|
|
.init(name: "简短直接", description: "反复使用短句", confidence: 0.7, supportCount: 3)
|
|
],
|
|
contradictions: []
|
|
),
|
|
reply: .init(traits: [], contradictions: []),
|
|
learnedThrough: Date(timeIntervalSince1970: 90),
|
|
cumulativeASRExampleCount: 5,
|
|
cumulativeASREffectiveCharacterCount: 2_500,
|
|
cumulativeReplyExampleCount: 0,
|
|
cumulativeReplyFinalEditCount: 0
|
|
)
|
|
|
|
private static func pack(id: String, generatedAt: Date) -> PolishStylePack {
|
|
PolishStylePack(
|
|
id: id,
|
|
name: "我的说话风格",
|
|
prompt: "# 角色\n自然\n# 风格边界\n自然\n# 示例\n输入 → 输出",
|
|
learningMetadata: PolishStylePack.LearningMetadata(
|
|
schemaVersion: 4,
|
|
evidenceStatus: "sufficient",
|
|
confidence: 0.8,
|
|
asrExampleCount: 5,
|
|
asrEffectiveCharacterCount: 2_500,
|
|
replyExampleCount: 0,
|
|
replyFinalEditCount: 0,
|
|
generatedAt: generatedAt
|
|
)
|
|
)
|
|
}
|
|
}
|