@n2flowjs/nbase/test/test-helpers/vector-db-test-utils.ts

Neural Vector Database for efficient similarity search

import * as fs from 'fs';
import * as path from 'path';
import { rimraf } from 'rimraf';
import { Vector } from '../../src/types';

export const TEST_DIR = path.join(process.cwd(), 'test-data', 'partitioned-db');
export const PARTITIONS_DIR = path.join(TEST_DIR, 'partitions');

// Test vectors
export const generateRandomVector = (size: number): number[] => {
    return Array.from({ length: size }, () => Math.random() - 0.5);
};

/**
 * Creates test vectors with stable IDs for testing
 */
export function createTestVectors(count: number, dimension: number): any[] {
  return Array.from({ length: count }).map((_, i) => ({
    id: `test-${i}`, // Generate stable string IDs
    vector: new Float32Array(
      Array.from({ length: dimension }).map(() => Math.random())
    ),
    metadata: {
      testIndex: i,
      source: 'test-generation',
      created: new Date().toISOString()
    }
  }));
}

// Setup and teardown helpers
export const setupTestDirectory = async () => {
    if (fs.existsSync(TEST_DIR)) {
        await rimraf(TEST_DIR);
    }
    fs.mkdirSync(TEST_DIR, { recursive: true });
};

export const cleanupPartitionsDir = async () => {
    await rimraf(PARTITIONS_DIR);
};

export const cleanupTestDirectory = async () => {
    await rimraf(TEST_DIR);
};