Improved tests

pgvector · Jul 14, 2024 · e76ed76 · e76ed76
1 parent 90765e5
commit e76ed76
Show file tree

Hide file tree

Showing 11 changed files with 13 additions and 12 deletions.
diff --git a/tests/drizzle-orm/index.test.mjs b/tests/drizzle-orm/index.test.mjs
@@ -6,7 +6,7 @@ import { pgTable, serial, vector, halfvec, bit, sparsevec } from 'drizzle-orm/pg
 import { SparseVector } from 'pgvector';
 import postgres from 'postgres';
 
-test('example', async () => {
+test('drizzle-orm example', async () => {
  const client = postgres({database: 'pgvector_node_test', onnotice: function () { }});
  const db = drizzle(client);
 

diff --git a/tests/knex/index.test.mjs b/tests/knex/index.test.mjs
@@ -4,7 +4,7 @@ import Knex from 'knex';
 import pgvector from 'pgvector/knex';
 import { SparseVector } from 'pgvector';
 
-test('example', async () => {
+test('knex example', async () => {
  const knex = Knex({
  client: 'pg',
  connection: {database: 'pgvector_node_test'}

diff --git a/tests/kysely/index.test.mjs b/tests/kysely/index.test.mjs
@@ -6,7 +6,7 @@ import pgvector from 'pgvector/kysely';
 import { l2Distance, maxInnerProduct, cosineDistance, l1Distance, hammingDistance, jaccardDistance } from 'pgvector/kysely';
 import { SparseVector } from 'pgvector';
 
-test('example', async () => {
+test('kysely example', async () => {
  const dialect = new PostgresDialect({
  pool: new pg.Pool({
  database: 'pgvector_node_test'

diff --git a/tests/mikro-orm/index.test.mjs b/tests/mikro-orm/index.test.mjs
@@ -4,7 +4,7 @@ import { MikroORM, EntityManager, EntitySchema } from '@mikro-orm/postgresql';
 import { VectorType, HalfvecType, BitType, SparsevecType, l2Distance, maxInnerProduct, cosineDistance, l1Distance, hammingDistance, jaccardDistance } from 'pgvector/mikro-orm';
 import { SparseVector } from 'pgvector';
 
-test('example', async () => {
+test('mikro-orm example', async () => {
  const Item = new EntitySchema({
  name: 'Item',
  tableName: 'mikro_items',
@@ -25,9 +25,10 @@ test('example', async () => {
  const em = orm.em.fork();
 
  await em.execute('CREATE EXTENSION IF NOT EXISTS vector');
+ await em.execute('DROP TABLE IF EXISTS mikro_items');
 
  const generator = orm.getSchemaGenerator();
- await generator.refreshDatabase();
+ await generator.createSchema();
 
  em.create(Item, {embedding: [1, 1, 1], half_embedding: [1, 1, 1], binary_embedding: '000', sparse_embedding: new SparseVector([1, 1, 1])});
  em.create(Item, {embedding: [2, 2, 2], half_embedding: [2, 2, 2], binary_embedding: '101', sparse_embedding: new SparseVector([2, 2, 2])});

diff --git a/tests/objection/index.test.mjs b/tests/objection/index.test.mjs
@@ -6,7 +6,7 @@ import pgvector from 'pgvector/objection';
 import { l2Distance, maxInnerProduct, cosineDistance, l1Distance, hammingDistance, jaccardDistance } from 'pgvector/objection';
 import { SparseVector } from 'pgvector';
 
-test('example', async () => {
+test('objection example', async () => {
  const knex = Knex({
  client: 'pg',
  connection: {database: 'pgvector_node_test'}

diff --git a/tests/pg-promise/index.test.mjs b/tests/pg-promise/index.test.mjs
@@ -4,7 +4,7 @@ import pgpromise from 'pg-promise';
 import pgvector from 'pgvector/pg-promise';
 import { SparseVector } from 'pgvector';
 
-test('example', async () => {
+test('pg-promise example', async () => {
  const initOptions = {
  async connect(e) {
  await pgvector.registerTypes(e.client);

diff --git a/tests/pg/index.test.mjs b/tests/pg/index.test.mjs
@@ -4,7 +4,7 @@ import pg from 'pg';
 import pgvector from 'pgvector/pg';
 import { SparseVector } from 'pgvector';
 
-test('example', async () => {
+test('pg example', async () => {
  const client = new pg.Client({database: 'pgvector_node_test'});
  await client.connect();
 

diff --git a/tests/postgres/index.test.mjs b/tests/postgres/index.test.mjs
@@ -4,7 +4,7 @@ import postgres from 'postgres';
 import pgvector from 'pgvector';
 import { SparseVector } from 'pgvector';
 
-test('example', async () => {
+test('postgres example', async () => {
  const sql = postgres({database: 'pgvector_node_test', onnotice: function () { }});
 
  await sql`CREATE EXTENSION IF NOT EXISTS vector`;

diff --git a/tests/sequelize/index.test.mjs b/tests/sequelize/index.test.mjs
@@ -5,7 +5,7 @@ import pgvector from 'pgvector/sequelize';
 import { l2Distance, maxInnerProduct, cosineDistance, l1Distance, hammingDistance, jaccardDistance } from 'pgvector/sequelize';
 import { SparseVector } from 'pgvector';
 
-test('example', async () => {
+test('sequelize example', async () => {
  pgvector.registerTypes(Sequelize);
 
  let sequelize = new Sequelize('postgres://localhost/pgvector_node_test', {

diff --git a/tests/slonik/index.test.mjs b/tests/slonik/index.test.mjs
@@ -4,7 +4,7 @@ import pgvector from 'pgvector';
 import { SparseVector } from 'pgvector';
 import { createPool, sql } from 'slonik';
 
-test('example', async () => {
+test('slonik example', async () => {
  const pool = await createPool('postgres://localhost/pgvector_node_test');
 
  await pool.query(sql.unsafe`CREATE EXTENSION IF NOT EXISTS vector`);

diff --git a/tests/typeorm/index.test.mjs b/tests/typeorm/index.test.mjs
@@ -4,7 +4,7 @@ import pgvector from 'pgvector';
 import { SparseVector } from 'pgvector';
 import { DataSource, EntitySchema } from 'typeorm';
 
-test('example', async () => {
+test('typeorm example', async () => {
  // entity definition without decorators
  // https://typeorm.io/separating-entity-definition
  const Item = new EntitySchema({