Skip to content

Commit d10751e

Browse files
committed
[fix] Fix immediate_downstream/upstream UNSET assertions in lineage tests
Change `is None` to `not` for immediate_downstream and immediate_upstream checks to handle msgspec UNSET values in v9 lineage test assertions. Made-with: Cursor
1 parent 76c5b17 commit d10751e

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests_v9/integration/lineage_test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -519,7 +519,7 @@ def test_fetch_lineage_start_list_detailed(
519519
assert results[4].guid == view.guid
520520
assert results[4].immediate_upstream and len(results[4].immediate_upstream) == 1
521521
assert results[4].immediate_upstream[0].guid == mview.guid
522-
assert results[4].immediate_downstream is None
522+
assert not results[4].immediate_downstream
523523
lineage = FluentLineage(
524524
starting_guid=table.guid,
525525
direction=LineageDirection.UPSTREAM,
@@ -529,8 +529,8 @@ def test_fetch_lineage_start_list_detailed(
529529
assert response
530530
assert not response.has_more
531531
assets = response.current_page()
532-
assert assets[0].immediate_upstream is None
533-
assert assets[0].immediate_downstream is None
532+
assert not assets[0].immediate_upstream
533+
assert not assets[0].immediate_downstream
534534

535535

536536
def test_fetch_lineage_middle_list(

0 commit comments

Comments
 (0)