From 51e9381854b26dd6552e02d90d62448e8fe4f208 Mon Sep 17 00:00:00 2001 From: Rob Howley Date: Fri, 25 Oct 2024 16:55:00 -0400 Subject: [PATCH] can we force separate environments? Signed-off-by: Rob Howley --- .../online_store/test_universal_online.py | 22 ++++++++++++++----- 1 file changed, 17 insertions(+), 5 deletions(-) diff --git a/sdk/python/tests/integration/online_store/test_universal_online.py b/sdk/python/tests/integration/online_store/test_universal_online.py index c543645753..a5493fbdb1 100644 --- a/sdk/python/tests/integration/online_store/test_universal_online.py +++ b/sdk/python/tests/integration/online_store/test_universal_online.py @@ -485,11 +485,7 @@ def test_online_retrieval_with_event_timestamps(environment, universal_data_sour assert_feature_store_universal_feature_views_response(df) -@pytest.mark.integration -@pytest.mark.universal_online_stores(only=["redis", "dynamodb", "postgres"]) -async def test_async_online_retrieval_with_event_timestamps( - environment, universal_data_sources -): +async def _do_async_retrieval_test(environment, universal_data_sources): fs = setup_feature_store_universal_feature_views( environment, universal_data_sources ) @@ -510,6 +506,22 @@ async def test_async_online_retrieval_with_event_timestamps( await fs.close() +@pytest.mark.integration +@pytest.mark.universal_online_stores(only=["redis", "postgres"]) +async def test_async_online_retrieval_with_event_timestamps( + environment, universal_data_sources +): + await _do_async_retrieval_test(environment, universal_data_sources) + + +@pytest.mark.integration +@pytest.mark.universal_online_stores(only=["dynamodb"]) +async def test_async_online_retrieval_with_event_timestamps_dynamo( + environment, universal_data_sources +): + await _do_async_retrieval_test(environment, universal_data_sources) + + @pytest.mark.integration @pytest.mark.universal_online_stores def test_online_list_retrieval(environment, universal_data_sources):