fix: Resolve soft delete and add encrypted config support
- Fix soft delete functionality by filtering out deleted items in search results - Add support for encrypted configuration storage and retrieval - Copy test suite and vitest config to ensure all functionality works - Maintain all core brainy functionality during repository cleanup
This commit is contained in:
parent
7432fab3bc
commit
3d80df1726
62 changed files with 13792 additions and 6 deletions
140
tests/statistics.test.ts
Normal file
140
tests/statistics.test.ts
Normal file
|
|
@ -0,0 +1,140 @@
|
|||
/**
|
||||
* Statistics Functionality Tests
|
||||
* Tests the getStatistics function as a consumer would use it
|
||||
*/
|
||||
|
||||
import { describe, it, expect, beforeAll } from 'vitest'
|
||||
|
||||
/**
|
||||
* Helper function to create a 512-dimensional vector for testing
|
||||
* @param primaryIndex The index to set to 1.0, all other indices will be 0.0
|
||||
* @returns A 512-dimensional vector with a single 1.0 value at the specified index
|
||||
*/
|
||||
function createTestVector(primaryIndex: number = 0): number[] {
|
||||
const vector = new Array(384).fill(0)
|
||||
vector[primaryIndex % 512] = 1.0
|
||||
return vector
|
||||
}
|
||||
|
||||
describe('Brainy Statistics Functionality', () => {
|
||||
let brainy: any
|
||||
|
||||
beforeAll(async () => {
|
||||
// Load brainy library as a consumer would
|
||||
brainy = await import('../dist/unified.js')
|
||||
})
|
||||
|
||||
describe('Library Exports', () => {
|
||||
it('should export getStatistics function at the root level', () => {
|
||||
expect(brainy.getStatistics).toBeDefined()
|
||||
expect(typeof brainy.getStatistics).toBe('function')
|
||||
})
|
||||
})
|
||||
|
||||
describe('getStatistics Functionality', () => {
|
||||
it('should retrieve statistics from a BrainyData instance', async () => {
|
||||
// Create a BrainyData instance
|
||||
const data = new brainy.BrainyData({
|
||||
metric: 'euclidean'
|
||||
})
|
||||
|
||||
await data.init()
|
||||
await data.clear() // Clear any existing data
|
||||
|
||||
// Add some test data
|
||||
await data.add(createTestVector(0), { id: 'v1', label: 'x-axis' })
|
||||
await data.add(createTestVector(1), { id: 'v2', label: 'y-axis' })
|
||||
await data.add(createTestVector(2), { id: 'v3', label: 'z-axis' })
|
||||
|
||||
// Add a verb
|
||||
await data.addVerb('v1', 'v2', createTestVector(3), { type: 'connected_to' })
|
||||
|
||||
// Get statistics using the standalone function
|
||||
const stats = await brainy.getStatistics(data)
|
||||
|
||||
// Verify statistics
|
||||
expect(stats).toBeDefined()
|
||||
expect(stats.nounCount).toBe(3)
|
||||
expect(stats.verbCount).toBe(1)
|
||||
expect(stats.metadataCount).toBe(3) // Each noun has metadata
|
||||
expect(stats.hnswIndexSize).toBe(4) // 3 nouns + 1 verb (verbs are also added to HNSW index)
|
||||
})
|
||||
|
||||
it('should throw an error when no instance is provided', async () => {
|
||||
await expect(brainy.getStatistics()).rejects.toThrow('BrainyData instance must be provided')
|
||||
})
|
||||
|
||||
it('should match the instance method results', async () => {
|
||||
// Create a BrainyData instance
|
||||
const data = new brainy.BrainyData({})
|
||||
|
||||
await data.init()
|
||||
|
||||
// Add some test data
|
||||
await data.add(createTestVector(5), { id: 'test1' })
|
||||
|
||||
// Get statistics using both methods
|
||||
const instanceStats = await data.getStatistics()
|
||||
const functionStats = await brainy.getStatistics(data)
|
||||
|
||||
// Verify core statistics match (ignoring volatile fields like memoryUsage and timestamps)
|
||||
expect(functionStats.nounCount).toBe(instanceStats.nounCount)
|
||||
expect(functionStats.verbCount).toBe(instanceStats.verbCount)
|
||||
expect(functionStats.metadataCount).toBe(instanceStats.metadataCount)
|
||||
expect(functionStats.hnswIndexSize).toBe(instanceStats.hnswIndexSize)
|
||||
|
||||
// If serviceBreakdown exists, verify it matches
|
||||
if (instanceStats.serviceBreakdown) {
|
||||
expect(functionStats.serviceBreakdown).toEqual(instanceStats.serviceBreakdown)
|
||||
}
|
||||
})
|
||||
|
||||
it('should track statistics by service', async () => {
|
||||
// Create a BrainyData instance
|
||||
const data = new brainy.BrainyData({
|
||||
metric: 'euclidean'
|
||||
})
|
||||
|
||||
await data.init()
|
||||
await data.clear() // Clear any existing data
|
||||
|
||||
// Add data from different services
|
||||
await data.add(createTestVector(10), { id: 'v1', label: 'service1-item' }, { service: 'service1' })
|
||||
await data.add(createTestVector(20), { id: 'v2', label: 'service1-item' }, { service: 'service1' })
|
||||
await data.add(createTestVector(30), { id: 'v3', label: 'service2-item' }, { service: 'service2' })
|
||||
|
||||
// Add verbs from different services
|
||||
await data.addVerb('v1', 'v2', undefined, { type: 'related_to', service: 'service1' })
|
||||
await data.addVerb('v2', 'v3', undefined, { type: 'related_to', service: 'service2' })
|
||||
|
||||
// Get statistics for all services
|
||||
const allStats = await data.getStatistics()
|
||||
|
||||
// Verify total counts
|
||||
expect(allStats.nounCount).toBe(3)
|
||||
expect(allStats.verbCount).toBe(2)
|
||||
expect(allStats.metadataCount).toBe(3)
|
||||
|
||||
// Verify service breakdown exists
|
||||
expect(allStats.serviceBreakdown).toBeDefined()
|
||||
|
||||
// Verify service1 statistics
|
||||
const service1Stats = await data.getStatistics({ service: 'service1' })
|
||||
expect(service1Stats.nounCount).toBe(2)
|
||||
expect(service1Stats.verbCount).toBe(1)
|
||||
expect(service1Stats.metadataCount).toBe(2)
|
||||
|
||||
// Verify service2 statistics
|
||||
const service2Stats = await data.getStatistics({ service: 'service2' })
|
||||
expect(service2Stats.nounCount).toBe(1)
|
||||
expect(service2Stats.verbCount).toBe(1)
|
||||
expect(service2Stats.metadataCount).toBe(1)
|
||||
|
||||
// Verify multiple services filter
|
||||
const combinedStats = await data.getStatistics({ service: ['service1', 'service2'] })
|
||||
expect(combinedStats.nounCount).toBe(3)
|
||||
expect(combinedStats.verbCount).toBe(2)
|
||||
expect(combinedStats.metadataCount).toBe(3)
|
||||
})
|
||||
})
|
||||
})
|
||||
Loading…
Add table
Add a link
Reference in a new issue