Skip to content

Commit bef3fca

Browse files
committed
Improved typechecking for tests [skip ci]
1 parent 950604f commit bef3fca

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

tests/drizzle-orm.test.js

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,12 +23,11 @@ test('drizzle-orm example', async () => {
2323
});
2424

2525
const newItems = [
26-
{embedding: [1, 1, 1], halfEmbedding: [1, 1, 1], binaryEmbedding: '000', sparseEmbedding: new SparseVector([1, 1, 1])},
27-
{embedding: [2, 2, 2], halfEmbedding: [2, 2, 2], binaryEmbedding: '101', sparseEmbedding: new SparseVector([2, 2, 2])},
28-
{embedding: [1, 1, 2], halfEmbedding: [1, 1, 2], binaryEmbedding: '111', sparseEmbedding: new SparseVector([1, 1, 2])},
26+
{embedding: [1, 1, 1], halfEmbedding: [1, 1, 1], binaryEmbedding: '000', sparseEmbedding: (new SparseVector([1, 1, 1])).toString()},
27+
{embedding: [2, 2, 2], halfEmbedding: [2, 2, 2], binaryEmbedding: '101', sparseEmbedding: (new SparseVector([2, 2, 2])).toString()},
28+
{embedding: [1, 1, 2], halfEmbedding: [1, 1, 2], binaryEmbedding: '111', sparseEmbedding: (new SparseVector([1, 1, 2])).toString()},
2929
{embedding: null}
3030
];
31-
// @ts-ignore
3231
await db.insert(items).values(newItems);
3332

3433
// L2 distance
@@ -51,8 +50,7 @@ test('drizzle-orm example', async () => {
5150
// L2 distance - sparsevec
5251
allItems = await db.select()
5352
.from(items)
54-
// @ts-ignore
55-
.orderBy(l2Distance(items.sparseEmbedding, new SparseVector([1, 1, 1])))
53+
.orderBy(l2Distance(items.sparseEmbedding, (new SparseVector([1, 1, 1])).toString()))
5654
.limit(5);
5755
assert.deepEqual(allItems.map(v => v.id), [1, 3, 2, 4]);
5856

0 commit comments

Comments
 (0)