From 3869d1c4cf0b0190dd6df98bfb7c872a622eda6f Mon Sep 17 00:00:00 2001 From: Tommy Smith Date: Fri, 26 Sep 2025 15:59:35 +0100 Subject: [PATCH] Use `!` to simplify test logic --- src/collections/vectors/journey.test.ts | 57 +++++++++---------------- 1 file changed, 21 insertions(+), 36 deletions(-) diff --git a/src/collections/vectors/journey.test.ts b/src/collections/vectors/journey.test.ts index 1f75ce2a..b5230fd3 100644 --- a/src/collections/vectors/journey.test.ts +++ b/src/collections/vectors/journey.test.ts @@ -1,12 +1,7 @@ /* eslint-disable @typescript-eslint/no-non-null-assertion */ /* eslint-disable @typescript-eslint/no-non-null-asserted-optional-chain */ import { requireAtLeast } from '../../../test/version.js'; -import weaviate, { - VectorIndexConfigHNSW, - WeaviateClient, - WeaviateField, - WeaviateGenericObject, -} from '../../index.js'; +import weaviate, { VectorIndexConfigHNSW, WeaviateClient } from '../../index.js'; import { Collection } from '../collection/index.js'; import { MultiVectorType, SingleVectorType } from '../query/types.js'; @@ -78,40 +73,30 @@ requireAtLeast(1, 29, 0).describe( it('should be able to get the inserted object with its vectors stated implicitly', async () => { const obj = await collection.query.fetchObjectById(id1, { includeVector: true }); - const assert = (obj: any): obj is WeaviateGenericObject, MyVectors> => { - expect(obj).not.toBeNull(); - return true; - }; - if (assert(obj)) { - singleVector = obj.vectors.regular; - multiVector = obj.vectors.colbert; - expect(obj.uuid).toBe(id1); - expect(obj.vectors).toBeDefined(); - expect(obj.vectors.regular).toEqual([1, 2, 3, 4]); - expect(obj.vectors.colbert).toEqual([ - [1, 2], - [3, 4], - ]); - } + expect(obj).not.toBeNull(); + singleVector = obj!.vectors.regular; + multiVector = obj!.vectors.colbert; + expect(obj!.uuid).toBe(id1); + expect(obj!.vectors).toBeDefined(); + expect(obj!.vectors.regular).toEqual([1, 2, 3, 4]); + expect(obj!.vectors.colbert).toEqual([ + [1, 2], + [3, 4], + ]); }); it('should be able to get the inserted object with its vectors stated explicitly', async () => { const obj = await collection.query.fetchObjectById(id1, { includeVector: ['regular', 'colbert'] }); - const assert = (obj: any): obj is WeaviateGenericObject, MyVectors> => { - expect(obj).not.toBeNull(); - return true; - }; - if (assert(obj)) { - singleVector = obj.vectors.regular; - multiVector = obj.vectors.colbert; - expect(obj.uuid).toBe(id1); - expect(obj.vectors).toBeDefined(); - expect(obj.vectors.regular).toEqual([1, 2, 3, 4]); - expect(obj.vectors.colbert).toEqual([ - [1, 2], - [3, 4], - ]); - } + expect(obj).not.toBeNull(); + singleVector = obj!.vectors.regular; + multiVector = obj!.vectors.colbert; + expect(obj!.uuid).toBe(id1); + expect(obj!.vectors).toBeDefined(); + expect(obj!.vectors.regular).toEqual([1, 2, 3, 4]); + expect(obj!.vectors.colbert).toEqual([ + [1, 2], + [3, 4], + ]); }); it('should be able to get the inserted object with one of its vectors', async () => {