diff --git a/test/embedding/test_pgvector.py b/test/embedding/test_pgvector.py index 3e06c06..d1b6f0e 100644 --- a/test/embedding/test_pgvector.py +++ b/test/embedding/test_pgvector.py @@ -181,16 +181,12 @@ async def test_PGv_search_filtered(): assert sim_search is not None, Exception("No results returned from filtered search") #Test conjunctive semantics - await vDB.insert(content='Foo', title='Some text', page_numbers=[1], tags=['tag1']) - await vDB.insert(content='Baz', title='Some mo text', page_numbers=[1], tags=['tag2', 'tag3']) - await vDB.insert(content='Bar', title='Even mo text', page_numbers=[1], tags=['tag3']) + await vDB.insert(content='Text', title='Some text', page_numbers=[1], tags=['tag1']) + await vDB.insert(content='Text', title='Some mo text', page_numbers=[1], tags=['tag2', 'tag3']) + await vDB.insert(content='Text', title='Even mo text', page_numbers=[1], tags=['tag3']) - sim_search = await vDB.search(query_string='text', query_tags=['tag1', 'tag3'], conjunctive=False) + sim_search = await vDB.search(query_string='Text', query_tags=['tag1', 'tag3'], conjunctive=False) assert sim_search is not None, Exception("No results returned from filtered search") - assert len(sim_search) == 3, Exception("There should be 3 results") - - sim_search = await vDB.search(query_string='text', query_tags=['tag1'], conjunctive=False) - assert sim_search is not None, Exception("No results returned from filtered search") - assert len(sim_search) == 1, Exception("There should be 1 result") + assert len(sim_search) == 3, Exception(f"There should be 3 results, received {sim_search}") await vDB.drop_table()