summaryrefslogtreecommitdiffstats
path: root/tests/test_lineage.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_lineage.py')
-rw-r--r--tests/test_lineage.py7
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")