17 lines
492 B
Python
17 lines
492 B
Python
|
import unittest
|
||
|
|
|
||
|
|
from sqlglot import parse_one
|
||
|
|
from sqlglot.transforms import unalias_group
|
||
|
|
|
||
|
|
|
||
|
|
class TestTime(unittest.TestCase):
|
||
|
|
def validate(self, transform, sql, target):
|
||
|
|
self.assertEqual(parse_one(sql).transform(transform).sql(), target)
|
||
|
|
|
||
|
|
def test_unalias_group(self):
|
||
|
|
self.validate(
|
||
|
|
unalias_group,
|
||
|
|
"SELECT a, b AS b, c AS c, 4 FROM x GROUP BY a, b, x.c, 4",
|
||
|
|
"SELECT a, b AS b, c AS c, 4 FROM x GROUP BY a, 2, x.c, 4",
|
||
|
|
)
|