diff --git a/tests/cross_encoder/test_bge_reranker_client_int.py b/tests/cross_encoder/test_bge_reranker_client_int.py index b449d2e7..12aa965d 100644 --- a/tests/cross_encoder/test_bge_reranker_client_int.py +++ b/tests/cross_encoder/test_bge_reranker_client_int.py @@ -18,8 +18,6 @@ import pytest from graphiti_core.cross_encoder.bge_reranker_client import BGERerankerClient -pytestmark = pytest.mark.integration - @pytest.fixture def client(): @@ -27,7 +25,6 @@ def client(): @pytest.mark.asyncio -@pytest.mark.integration async def test_rank_basic_functionality(client): query = 'What is the capital of France?' passages = [ @@ -56,7 +53,6 @@ async def test_rank_basic_functionality(client): @pytest.mark.asyncio -@pytest.mark.integration async def test_rank_empty_input(client): query = 'Empty test' passages = [] @@ -68,7 +64,6 @@ async def test_rank_empty_input(client): @pytest.mark.asyncio -@pytest.mark.integration async def test_rank_single_passage(client): query = 'Test query' passages = ['Single test passage'] diff --git a/tests/driver/test_falkordb_driver.py b/tests/driver/test_falkordb_driver.py index 1220101a..7edf9afb 100644 --- a/tests/driver/test_falkordb_driver.py +++ b/tests/driver/test_falkordb_driver.py @@ -366,7 +366,6 @@ class TestDatetimeConversion: class TestFalkorDriverIntegration: """Simple integration test for FalkorDB driver.""" - @pytest.mark.integration @pytest.mark.asyncio @unittest.skipIf(not HAS_FALKORDB, 'FalkorDB is not installed') async def test_basic_integration_with_real_falkordb(self): diff --git a/tests/test_edge_int.py b/tests/test_edge_int.py index b028e452..15555d72 100644 --- a/tests/test_edge_int.py +++ b/tests/test_edge_int.py @@ -26,7 +26,6 @@ from graphiti_core.nodes import CommunityNode, EntityNode, EpisodeType, Episodic from tests.helpers_test import get_edge_count, get_node_count, group_id pytest_plugins = ('pytest_asyncio',) -pytestmark = pytest.mark.integration def setup_logging(): diff --git a/tests/test_entity_exclusion_int.py b/tests/test_entity_exclusion_int.py index 0ac9897c..53982bc9 100644 --- a/tests/test_entity_exclusion_int.py +++ b/tests/test_entity_exclusion_int.py @@ -23,7 +23,6 @@ from graphiti_core.graphiti import Graphiti from graphiti_core.helpers import validate_excluded_entity_types from tests.helpers_test import drivers, get_driver -pytestmark = pytest.mark.integration pytest_plugins = ('pytest_asyncio',) diff --git a/tests/test_graphiti_int.py b/tests/test_graphiti_int.py index 90033f35..766828e3 100644 --- a/tests/test_graphiti_int.py +++ b/tests/test_graphiti_int.py @@ -25,7 +25,6 @@ from graphiti_core.search.search_helpers import search_results_to_context_string from graphiti_core.utils.datetime_utils import utc_now from tests.helpers_test import GraphProvider -pytestmark = pytest.mark.integration pytest_plugins = ('pytest_asyncio',) diff --git a/tests/test_node_int.py b/tests/test_node_int.py index bdb8add8..7e73b856 100644 --- a/tests/test_node_int.py +++ b/tests/test_node_int.py @@ -33,8 +33,6 @@ from tests.helpers_test import ( group_id, ) -pytestmark = pytest.mark.integration - created_at = datetime.now() deleted_at = created_at + timedelta(days=3) valid_at = created_at + timedelta(days=1)