[Leon-Algo]

Summary

• Convert BFS/DFS/ECE partitioner integration tests from broken @pytest.mark.asyncio coroutine calls to synchronous calls matching the current synchronous BasePartitioner/storage APIs.
• Fix the invalid from graphgen.models import NetworkXStorage import (storage lives in graphgen.storage); these tests fail to even collect on main today.
• No production code changes; all 12 partitioner integration tests pass.

This is the first of three PRs split from #192 per review feedback (smallest first). PR B (prompt profiles/gates/concurrency/LLM patches) and PR C (event-entity pipeline) follow separately.

Validation

• pytest tests/integration_tests/models/partitioner/ → 12 passed.

🤖 Generated with Claude Code

[Leon-Algo]

Retriggering CI — the third-party translate action failed with a transient HTTP error (same workflow passes on #194/#195).

[Leon-Algo]

Friendly ping for review when you have a moment 🙂 The three-way split from #192 is ready and stable:

• Fix broken async partitioner integration tests #193 (this one) — partitioner test fixes only, no production code changes.
• Add versioned prompt profiles, output gates, bounded generation concurrency and env-gated LLM client patches #194 — prompt-profile mechanism + gate registry + llm_concurrency + env-gated LLM patches; no domain assets; pylint 10.00/10.
• Add opt-in event_entity KG builder, event_join partitioner and event extraction profiles #195 — opt-in event_entity/event_join pipeline; fixes the three functional bugs from your review (external profile path reload, -- support citations, tuple prefill vs JSON parser, plus membership role persistence).

CI note: the earlier translate red X here was a transient HTTPError from the third-party translate action (it passes on #194/#195 and on rerun); the branch head was advanced to retrigger it. Happy to adjust anything — thanks!

[ChenZiHong-Gavin]

LGTM. Pure test-side change: the partitioner/storage APIs are synchronous on main, the NetworkXStorage import now points at graphgen.storage, and the list() wrapping is correct for the generator-returning partitioners. Pylint CI is green on 3.10/3.11/3.12.

[Leon-Algo]

Thanks for reviewing and merging, @ChenZiHong-Gavin! Glad this helps keep the partitioner integration tests healthy on main. It felt good to hand over tests that run green against upstream APIs — thanks for making that easy.