Skip to content

Commit b125cae

Browse files
ppiegazeclaude
andcommitted
Enable GPU and stress tests, update skip reasons for clarity
- Remove skip from GPU tests (backend has GPU resources) - Remove skip from stress tests (ok to run) - Update skip reasons for external service and secrets tests to clarify they require Union backend configuration Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com> Signed-off-by: Peeter Piegaze <1153481+ppiegaze@users.noreply.github.com>
1 parent fae9de2 commit b125cae

File tree

1 file changed

+18
-27
lines changed

1 file changed

+18
-27
lines changed

examples/integration_tests.py

Lines changed: 18 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -562,127 +562,122 @@ async def test_apps_single_script_fastapi(flyte_client):
562562

563563

564564
# =============================================================================
565-
# GPU TESTS (skipped by default)
565+
# GPU TESTS
566566
# =============================================================================
567567

568568

569569
@pytest.mark.integration
570570
@pytest.mark.gpu
571-
@pytest.mark.skip(reason="Requires GPU resources - enable in GPU CI")
572571
@pytest.mark.asyncio
573572
async def test_accelerators_gpu(flyte_client):
574-
"""Test GPU accelerator example - requires GPU."""
573+
"""Test GPU accelerator example."""
575574
from examples.accelerators.gpu import main
576575

577576
await _run_and_wait(main, "test_accelerators_gpu")
578577

579578

580579
@pytest.mark.integration
581580
@pytest.mark.gpu
582-
@pytest.mark.skip(reason="Requires GPU resources - enable in GPU CI")
583581
@pytest.mark.asyncio
584582
async def test_ml_gridsearch_gpu(flyte_client):
585-
"""Test ML gridsearch with GPU - requires GPU."""
583+
"""Test ML gridsearch with GPU."""
586584
from examples.ml.gridsearch_gpu import main
587585

588586
await _run_and_wait(main, "test_ml_gridsearch_gpu")
589587

590588

591589
@pytest.mark.integration
592590
@pytest.mark.gpu
593-
@pytest.mark.skip(reason="Requires GPU resources - enable in GPU CI")
594591
@pytest.mark.asyncio
595592
async def test_genai_vllm_app(flyte_client):
596-
"""Test vLLM app deployment - requires GPU."""
593+
"""Test vLLM app deployment."""
597594
from examples.genai.vllm.vllm_app import app
598595

599596
await _deploy_and_verify(app, "test_genai_vllm_app")
600597

601598

602599
@pytest.mark.integration
603600
@pytest.mark.gpu
604-
@pytest.mark.skip(reason="Requires GPU resources - enable in GPU CI")
605601
@pytest.mark.asyncio
606602
async def test_genai_sglang_app(flyte_client):
607-
"""Test SGLang app deployment - requires GPU."""
603+
"""Test SGLang app deployment."""
608604
from examples.genai.sglang.sglang_app import app
609605

610606
await _deploy_and_verify(app, "test_genai_sglang_app")
611607

612608

613609
# =============================================================================
614-
# EXTERNAL SERVICE TESTS (skipped by default)
610+
# EXTERNAL SERVICE TESTS (skipped - require backend connector configuration)
615611
# =============================================================================
616612

617613

618614
@pytest.mark.integration
619615
@pytest.mark.external
620-
@pytest.mark.skip(reason="Requires external service configuration")
616+
@pytest.mark.skip(reason="Requires BigQuery connector configured in Union backend")
621617
@pytest.mark.asyncio
622618
async def test_connectors_bigquery(flyte_client):
623-
"""Test BigQuery connector - requires BigQuery access."""
619+
"""Test BigQuery connector."""
624620
from examples.connectors.bigquery_example import main
625621

626622
await _run_and_wait(main, "test_connectors_bigquery")
627623

628624

629625
@pytest.mark.integration
630626
@pytest.mark.external
631-
@pytest.mark.skip(reason="Requires external service configuration")
627+
@pytest.mark.skip(reason="Requires Databricks connector configured in Union backend")
632628
@pytest.mark.asyncio
633629
async def test_connectors_databricks(flyte_client):
634-
"""Test Databricks connector - requires Databricks access."""
630+
"""Test Databricks connector."""
635631
from examples.connectors.databricks_example import main
636632

637633
await _run_and_wait(main, "test_connectors_databricks")
638634

639635

640636
# =============================================================================
641-
# SECRETS TESTS (skipped by default)
637+
# SECRETS TESTS (skipped - require secrets configured in Union backend)
642638
# =============================================================================
643639

644640

645641
@pytest.mark.integration
646642
@pytest.mark.secrets
647-
@pytest.mark.skip(reason="Requires secrets configuration")
643+
@pytest.mark.skip(reason="Requires OpenAI API key configured as secret in Union backend")
648644
@pytest.mark.asyncio
649645
async def test_genai_openai_agent(flyte_client):
650-
"""Test OpenAI agent - requires API key."""
646+
"""Test OpenAI agent."""
651647
from examples.genai.openai_agent import main
652648

653649
await _run_and_wait(main, "test_genai_openai_agent")
654650

655651

656652
@pytest.mark.integration
657653
@pytest.mark.secrets
658-
@pytest.mark.skip(reason="Requires secrets configuration")
654+
@pytest.mark.skip(reason="Requires OpenAI API key configured as secret in Union backend")
659655
@pytest.mark.asyncio
660656
async def test_genai_langgraph_agent(flyte_client):
661-
"""Test LangGraph agent - requires API key."""
657+
"""Test LangGraph agent."""
662658
from examples.genai.langgraph_agent import main
663659

664660
await _run_and_wait(main, "test_genai_langgraph_agent")
665661

666662

667663
@pytest.mark.integration
668664
@pytest.mark.secrets
669-
@pytest.mark.skip(reason="Requires secrets configuration")
665+
@pytest.mark.skip(reason="Requires W&B API key configured as secret in Union backend")
670666
@pytest.mark.asyncio
671667
async def test_context_wandb(flyte_client):
672-
"""Test W&B context - requires W&B API key."""
668+
"""Test W&B context."""
673669
from examples.context.wandb_context import main
674670

675671
await _run_and_wait(main, "test_context_wandb")
676672

677673

678674
# =============================================================================
679-
# STRESS TESTS (skipped by default)
675+
# STRESS TESTS
680676
# =============================================================================
681677

682678

683679
@pytest.mark.integration
684680
@pytest.mark.stress
685-
@pytest.mark.skip(reason="Stress test - enable manually")
686681
@pytest.mark.asyncio
687682
async def test_stress_large_fanout(flyte_client):
688683
"""Test large fanout stress test."""
@@ -693,7 +688,6 @@ async def test_stress_large_fanout(flyte_client):
693688

694689
@pytest.mark.integration
695690
@pytest.mark.stress
696-
@pytest.mark.skip(reason="Stress test - enable manually")
697691
@pytest.mark.asyncio
698692
async def test_stress_long_running(flyte_client):
699693
"""Test long-running stress test."""
@@ -704,7 +698,6 @@ async def test_stress_long_running(flyte_client):
704698

705699
@pytest.mark.integration
706700
@pytest.mark.stress
707-
@pytest.mark.skip(reason="Stress test - enable manually")
708701
@pytest.mark.asyncio
709702
async def test_stress_runs_per_second(flyte_client):
710703
"""Test runs-per-second stress test."""
@@ -715,7 +708,6 @@ async def test_stress_runs_per_second(flyte_client):
715708

716709
@pytest.mark.integration
717710
@pytest.mark.stress
718-
@pytest.mark.skip(reason="Stress test - enable manually")
719711
@pytest.mark.asyncio
720712
async def test_stress_large_file_io(flyte_client):
721713
"""Test large file I/O stress test."""
@@ -726,7 +718,6 @@ async def test_stress_large_file_io(flyte_client):
726718

727719
@pytest.mark.integration
728720
@pytest.mark.stress
729-
@pytest.mark.skip(reason="Stress test - enable manually")
730721
@pytest.mark.asyncio
731722
async def test_stress_fanout_concurrency(flyte_client):
732723
"""Test fanout concurrency stress test."""

0 commit comments

Comments
 (0)