diff options
Diffstat (limited to 'tests/test_lineage.py')
-rw-r--r-- | tests/test_lineage.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/test_lineage.py b/tests/test_lineage.py index 8755b42..66b8b15 100644 --- a/tests/test_lineage.py +++ b/tests/test_lineage.py @@ -296,3 +296,10 @@ class TestLineage(unittest.TestCase): downstream = downstream.downstream[0] self.assertEqual(downstream.name, "*") self.assertEqual(downstream.source.sql(), "table_a AS table_a") + + def test_unnest(self) -> None: + node = lineage( + "b", + "with _data as (select [struct(1 as a, 2 as b)] as col) select b from _data cross join unnest(col)", + ) + self.assertEqual(node.name, "b") |