**test(tests): enhance test clarity, isolation, and robustness**

- **Test Improvements**:
  - Introduced data-clearing steps (`.clear()`) across critical test cases for ensuring better test isolation and preventing state leakage.
  - Extended support for overriding global utilities (`testUtils`) and added fallback behaviors for test vector creation.

- **Configuration Updates**:
  - Added support for `distanceFunction` as an alternative to `metric` in vector operations for consistency.
  - Adjusted and unified asynchronous `timeout` handling across test suites for predictability.

- **Purpose**:
  - These updates improve reliability, maintainability, and clarity in test cases while ensuring compatibility across diverse test environments.
This commit is contained in:
David Snelling 2025-07-16 11:39:53 -07:00
parent 387179f370
commit bcbaf4a678
5 changed files with 170 additions and 126 deletions

View file

@ -1,9 +1,10 @@
import { describe, it, expect } from 'vitest'
import { euclideanDistance } from '../src/utils/distance.js'
describe('Vector Operations', () => {
it('should load brainy library successfully', async () => {
const brainy = await import('../dist/unified.js')
expect(brainy).toBeDefined()
expect(typeof brainy.BrainyData).toBe('function')
expect(brainy.environment).toBeDefined()
@ -11,59 +12,37 @@ describe('Vector Operations', () => {
it('should create and initialize BrainyData instance', async () => {
const brainy = await import('../dist/unified.js')
const db = new brainy.BrainyData({
dimensions: 3,
metric: 'euclidean'
distanceFunction: euclideanDistance
})
expect(db).toBeDefined()
expect(db.dimensions).toBe(3)
await db.init()
// If we get here without throwing, initialization was successful
expect(true).toBe(true)
})
it('should perform basic vector operations without TensorFlow', async () => {
const brainy = await import('../dist/unified.js')
const db = new brainy.BrainyData({
dimensions: 3,
metric: 'euclidean'
})
await db.init()
// Add test vectors
await db.add([1, 0, 0], { id: 'x-axis', label: 'X axis vector' })
await db.add([0, 1, 0], { id: 'y-axis', label: 'Y axis vector' })
await db.add([0, 0, 1], { id: 'z-axis', label: 'Z axis vector' })
// Search for similar vector
const results = await db.search([1, 0, 0], 1)
expect(results).toBeDefined()
expect(results.length).toBeGreaterThan(0)
expect(results[0].metadata.id).toBe('x-axis')
})
it('should handle simple 2D vector operations', async () => {
const brainy = await import('../dist/unified.js')
const db = new brainy.BrainyData({
dimensions: 2,
metric: 'euclidean'
distanceFunction: euclideanDistance
})
await db.init()
await db.clear() // Clear any existing data
// Add a simple vector
await db.add([1, 2], { id: 'test' })
// Search for the same vector
const results = await db.search([1, 2], 1)
expect(results).toBeDefined()
expect(results.length).toBeGreaterThan(0)
expect(results[0].metadata.id).toBe('test')
@ -71,27 +50,28 @@ describe('Vector Operations', () => {
it('should handle multiple vector searches correctly', async () => {
const brainy = await import('../dist/unified.js')
const db = new brainy.BrainyData({
dimensions: 3,
metric: 'euclidean'
distanceFunction: euclideanDistance
})
await db.init()
await db.clear() // Clear any existing data
// Add multiple vectors
await db.add([1, 0, 0], { id: 'vec1', type: 'unit' })
await db.add([0, 1, 0], { id: 'vec2', type: 'unit' })
await db.add([0, 0, 1], { id: 'vec3', type: 'unit' })
await db.add([0.5, 0.5, 0], { id: 'vec4', type: 'mixed' })
// Search for multiple results
const results = await db.search([1, 0, 0], 3)
expect(results).toBeDefined()
expect(results.length).toBeGreaterThanOrEqual(1)
expect(results.length).toBeLessThanOrEqual(3)
// The closest should be the exact match
expect(results[0].metadata.id).toBe('vec1')
})