33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from tests.dialects.test_dialect import Validator
|
|
|
|
|
|
class TestDatabricks(Validator):
|
|
dialect = "databricks"
|
|
|
|
def test_datediff(self):
|
|
self.validate_all(
|
|
"SELECT DATEDIFF(year, 'start', 'end')",
|
|
write={
|
|
"tsql": "SELECT DATEDIFF(year, 'start', 'end')",
|
|
"databricks": "SELECT DATEDIFF(year, 'start', 'end')",
|
|
},
|
|
)
|
|
|
|
def test_add_date(self):
|
|
self.validate_all(
|
|
"SELECT DATEADD(year, 1, '2020-01-01')",
|
|
write={
|
|
"tsql": "SELECT DATEADD(year, 1, '2020-01-01')",
|
|
"databricks": "SELECT DATEADD(year, 1, '2020-01-01')",
|
|
},
|
|
)
|
|
self.validate_all(
|
|
"SELECT DATEDIFF('end', 'start')", write={"databricks": "SELECT DATEDIFF(DAY, 'start', 'end')"}
|
|
)
|
|
self.validate_all(
|
|
"SELECT DATE_ADD('2020-01-01', 1)",
|
|
write={
|
|
"tsql": "SELECT DATEADD(DAY, 1, '2020-01-01')",
|
|
"databricks": "SELECT DATEADD(DAY, 1, '2020-01-01')",
|
|
},
|
|
)
|