ColtenOuO opened a new pull request, #70042:
URL: https://github.com/apache/airflow/pull/70042
The `edges` module in the Task SDK provides `EdgeModifier` and its `Label`
factory — the API used to attach human-readable labels to task dependencies,
e.g. `t1 >> Label("branch a") >> t2`. Despite being user-facing, it had no
dedicated test module; it was only exercised indirectly through
operator-chaining tests, leaving several branches uncovered.
This PR adds `task-sdk/tests/task_sdk/definitions/test_edges.py` with
focused unit tests for the module's own behavior:
- `Label()` factory and `EdgeModifier` label handling (including the default
`None` label)
- the `roots` / `leaves` properties
- `_make_list` for single items vs. sequences
- the `_save_nodes` type guard raising `TypeError` for unsupported nodes
- `update_relative` for the upstream/downstream sides
- edge-info wiring through `>>` / `<<` (including fan-out to multiple
downstream tasks) and `add_edge_info`
- the `EdgeInfoType` shape
---
##### Was generative AI tooling used to co-author this PR?
- [X] Yes — Claude Code (Opus 4.8)
Generated-by: Claude Code (Opus 4.8) following [the
guidelines](https://github.com/apache/airflow/blob/main/contributing-docs/05_pull_requests.rst#gen-ai-assisted-contributions)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
