diff --git a/tests/test_graphiti_mock.py b/tests/test_graphiti_mock.py index 62a8ffd3..a30c9392 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 in [GraphProvider.FALKORDB, GraphProvider.KUZU]: - pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu') + if graph_driver.provider == GraphProvider.KUZU: + pytest.skip('Skipping as fulltext indexing not supported for 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 in [GraphProvider.FALKORDB, GraphProvider.KUZU]: - pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu') + if graph_driver.provider == GraphProvider.KUZU: + pytest.skip('Skipping as fulltext indexing not supported for 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 in [GraphProvider.FALKORDB, GraphProvider.KUZU]: - pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu') + if graph_driver.provider == GraphProvider.KUZU: + pytest.skip('Skipping as fulltext indexing not supported for 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 in [GraphProvider.FALKORDB, GraphProvider.KUZU]: - pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu') + if graph_driver.provider == GraphProvider.KUZU: + pytest.skip('Skipping as fulltext indexing not supported for Kuzu') graphiti = Graphiti( graph_driver=graph_driver,