(t *testing.T)
| 198 | } |
| 199 | |
| 200 | func TestInMemoryStore_GetTopK(t *testing.T) { |
| 201 | store := NewInMemoryStore() |
| 202 | defer func() { _ = store.Close() }() |
| 203 | |
| 204 | ctx := context.Background() |
| 205 | namespace := "user:123" |
| 206 | |
| 207 | for i := range 5 { |
| 208 | require.NoError(t, store.Save(ctx, &LogicMemory{ |
| 209 | Namespace: namespace, |
| 210 | Key: string(rune('a' + i)), |
| 211 | Provenance: &memory.MemoryProvenance{ |
| 212 | Confidence: float64(i+1) * 0.1, |
| 213 | }, |
| 214 | })) |
| 215 | } |
| 216 | |
| 217 | result, err := store.GetTopK(ctx, namespace, 3, OrderByConfidence) |
| 218 | require.NoError(t, err) |
| 219 | assert.Len(t, result, 3) |
| 220 | assert.InDelta(t, 0.5, result[0].Provenance.Confidence, 0.0001) |
| 221 | assert.InDelta(t, 0.4, result[1].Provenance.Confidence, 0.0001) |
| 222 | assert.InDelta(t, 0.3, result[2].Provenance.Confidence, 0.0001) |
| 223 | } |
| 224 | |
| 225 | func TestInMemoryStore_IncrementAccessCount(t *testing.T) { |
| 226 | store := NewInMemoryStore() |
nothing calls this directly
no test coverage detected