diff --git a/tests/test_graphiti_mock.py b/tests/test_graphiti_mock.py index f2ba89ec..62a8ffd3 100644 --- a/tests/test_graphiti_mock.py +++ b/tests/test_graphiti_mock.py @@ -918,8 +918,8 @@ async def test_get_communities_by_nodes(graph_driver, mock_embedder): async def test_edge_fulltext_search( graph_driver, mock_embedder, mock_llm_client, mock_cross_encoder_client ): - if graph_driver.provider == GraphProvider.FALKORDB: - pytest.skip('Skipping as tests fail on Falkordb') + if graph_driver.provider in [GraphProvider.FALKORDB, GraphProvider.KUZU]: + pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu') graphiti = Graphiti( graph_driver=graph_driver, @@ -1307,8 +1307,8 @@ async def test_edge_bfs_search(graph_driver, mock_embedder): async def test_node_fulltext_search( graph_driver, mock_embedder, mock_llm_client, mock_cross_encoder_client ): - if graph_driver.provider == GraphProvider.FALKORDB: - pytest.skip('Skipping as tests fail on Falkordb') + if graph_driver.provider in [GraphProvider.FALKORDB, GraphProvider.KUZU]: + pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu') graphiti = Graphiti( graph_driver=graph_driver, @@ -1516,8 +1516,8 @@ async def test_node_bfs_search(graph_driver, mock_embedder): async def test_episode_fulltext_search( graph_driver, mock_embedder, mock_llm_client, mock_cross_encoder_client ): - if graph_driver.provider == GraphProvider.FALKORDB: - pytest.skip('Skipping as tests fail on Falkordb') + if graph_driver.provider in [GraphProvider.FALKORDB, GraphProvider.KUZU]: + pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu') graphiti = Graphiti( graph_driver=graph_driver, @@ -1567,8 +1567,8 @@ async def test_episode_fulltext_search( async def test_community_fulltext_search( graph_driver, mock_embedder, mock_llm_client, mock_cross_encoder_client ): - if graph_driver.provider == GraphProvider.FALKORDB: - pytest.skip('Skipping as tests fail on Falkordb') + if graph_driver.provider in [GraphProvider.FALKORDB, GraphProvider.KUZU]: + pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu') graphiti = Graphiti( graph_driver=graph_driver,