Skip to content

Commit e76ed76

Browse files
committed
Improved tests
1 parent 90765e5 commit e76ed76

11 files changed

Lines changed: 13 additions & 12 deletions

File tree

tests/drizzle-orm/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import { pgTable, serial, vector, halfvec, bit, sparsevec } from 'drizzle-orm/pg
66
import { SparseVector } from 'pgvector';
77
import postgres from 'postgres';
88

9-
test('example', async () => {
9+
test('drizzle-orm example', async () => {
1010
const client = postgres({database: 'pgvector_node_test', onnotice: function () { }});
1111
const db = drizzle(client);
1212

tests/knex/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import Knex from 'knex';
44
import pgvector from 'pgvector/knex';
55
import { SparseVector } from 'pgvector';
66

7-
test('example', async () => {
7+
test('knex example', async () => {
88
const knex = Knex({
99
client: 'pg',
1010
connection: {database: 'pgvector_node_test'}

tests/kysely/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import pgvector from 'pgvector/kysely';
66
import { l2Distance, maxInnerProduct, cosineDistance, l1Distance, hammingDistance, jaccardDistance } from 'pgvector/kysely';
77
import { SparseVector } from 'pgvector';
88

9-
test('example', async () => {
9+
test('kysely example', async () => {
1010
const dialect = new PostgresDialect({
1111
pool: new pg.Pool({
1212
database: 'pgvector_node_test'

tests/mikro-orm/index.test.mjs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import { MikroORM, EntityManager, EntitySchema } from '@mikro-orm/postgresql';
44
import { VectorType, HalfvecType, BitType, SparsevecType, l2Distance, maxInnerProduct, cosineDistance, l1Distance, hammingDistance, jaccardDistance } from 'pgvector/mikro-orm';
55
import { SparseVector } from 'pgvector';
66

7-
test('example', async () => {
7+
test('mikro-orm example', async () => {
88
const Item = new EntitySchema({
99
name: 'Item',
1010
tableName: 'mikro_items',
@@ -25,9 +25,10 @@ test('example', async () => {
2525
const em = orm.em.fork();
2626

2727
await em.execute('CREATE EXTENSION IF NOT EXISTS vector');
28+
await em.execute('DROP TABLE IF EXISTS mikro_items');
2829

2930
const generator = orm.getSchemaGenerator();
30-
await generator.refreshDatabase();
31+
await generator.createSchema();
3132

3233
em.create(Item, {embedding: [1, 1, 1], half_embedding: [1, 1, 1], binary_embedding: '000', sparse_embedding: new SparseVector([1, 1, 1])});
3334
em.create(Item, {embedding: [2, 2, 2], half_embedding: [2, 2, 2], binary_embedding: '101', sparse_embedding: new SparseVector([2, 2, 2])});

tests/objection/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import pgvector from 'pgvector/objection';
66
import { l2Distance, maxInnerProduct, cosineDistance, l1Distance, hammingDistance, jaccardDistance } from 'pgvector/objection';
77
import { SparseVector } from 'pgvector';
88

9-
test('example', async () => {
9+
test('objection example', async () => {
1010
const knex = Knex({
1111
client: 'pg',
1212
connection: {database: 'pgvector_node_test'}

tests/pg-promise/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import pgpromise from 'pg-promise';
44
import pgvector from 'pgvector/pg-promise';
55
import { SparseVector } from 'pgvector';
66

7-
test('example', async () => {
7+
test('pg-promise example', async () => {
88
const initOptions = {
99
async connect(e) {
1010
await pgvector.registerTypes(e.client);

tests/pg/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import pg from 'pg';
44
import pgvector from 'pgvector/pg';
55
import { SparseVector } from 'pgvector';
66

7-
test('example', async () => {
7+
test('pg example', async () => {
88
const client = new pg.Client({database: 'pgvector_node_test'});
99
await client.connect();
1010

tests/postgres/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import postgres from 'postgres';
44
import pgvector from 'pgvector';
55
import { SparseVector } from 'pgvector';
66

7-
test('example', async () => {
7+
test('postgres example', async () => {
88
const sql = postgres({database: 'pgvector_node_test', onnotice: function () { }});
99

1010
await sql`CREATE EXTENSION IF NOT EXISTS vector`;

tests/sequelize/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ import pgvector from 'pgvector/sequelize';
55
import { l2Distance, maxInnerProduct, cosineDistance, l1Distance, hammingDistance, jaccardDistance } from 'pgvector/sequelize';
66
import { SparseVector } from 'pgvector';
77

8-
test('example', async () => {
8+
test('sequelize example', async () => {
99
pgvector.registerTypes(Sequelize);
1010

1111
let sequelize = new Sequelize('postgres://localhost/pgvector_node_test', {

tests/slonik/index.test.mjs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import pgvector from 'pgvector';
44
import { SparseVector } from 'pgvector';
55
import { createPool, sql } from 'slonik';
66

7-
test('example', async () => {
7+
test('slonik example', async () => {
88
const pool = await createPool('postgres://localhost/pgvector_node_test');
99

1010
await pool.query(sql.unsafe`CREATE EXTENSION IF NOT EXISTS vector`);

0 commit comments

Comments
 (0)