diff options
| author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-07-24 08:03:45 +0000 |
|---|---|---|
| committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-07-24 08:03:45 +0000 |
| commit | ca57dc468e5d8d0920e964d45ad25271ae6e633d (patch) | |
| tree | 319d8bffcb5c3e9afe9e62beca9ef401480578d2 /tests/test_lineage.py | |
| parent | Adding upstream version 17.4.1. (diff) | |
| download | sqlglot-ca57dc468e5d8d0920e964d45ad25271ae6e633d.tar.xz sqlglot-ca57dc468e5d8d0920e964d45ad25271ae6e633d.zip | |
Adding upstream version 17.7.0.upstream/17.7.0
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tests/test_lineage.py')
| -rw-r--r-- | tests/test_lineage.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/test_lineage.py b/tests/test_lineage.py index f33a2c2..0fd9da8 100644 --- a/tests/test_lineage.py +++ b/tests/test_lineage.py @@ -186,3 +186,16 @@ class TestLineage(unittest.TestCase): self.assertEqual(downstream.alias, "") self.assertEqual(downstream.downstream, []) + + def test_lineage_union(self) -> None: + node = lineage( + "x", + "SELECT ax AS x FROM a UNION SELECT bx FROM b UNION SELECT cx FROM c", + ) + assert len(node.downstream) == 3 + + node = lineage( + "x", + "SELECT x FROM (SELECT ax AS x FROM a UNION SELECT bx FROM b UNION SELECT cx FROM c)", + ) + assert len(node.downstream) == 3 |