Merging upstream version 18.7.0.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
77523b6777
commit
d1b976f442
96 changed files with 59037 additions and 52828 deletions
|
@ -95,16 +95,16 @@ class TestDataframeColumn(unittest.TestCase):
|
|||
self.assertEqual("cola IN (1, 2, 3)", F.col("cola").isin(1, 2, 3).sql())
|
||||
|
||||
def test_asc(self):
|
||||
self.assertEqual("cola", F.col("cola").asc().sql())
|
||||
self.assertEqual("cola ASC", F.col("cola").asc().sql())
|
||||
|
||||
def test_desc(self):
|
||||
self.assertEqual("cola DESC", F.col("cola").desc().sql())
|
||||
|
||||
def test_asc_nulls_first(self):
|
||||
self.assertEqual("cola", F.col("cola").asc_nulls_first().sql())
|
||||
self.assertEqual("cola ASC", F.col("cola").asc_nulls_first().sql())
|
||||
|
||||
def test_asc_nulls_last(self):
|
||||
self.assertEqual("cola NULLS LAST", F.col("cola").asc_nulls_last().sql())
|
||||
self.assertEqual("cola ASC NULLS LAST", F.col("cola").asc_nulls_last().sql())
|
||||
|
||||
def test_desc_nulls_first(self):
|
||||
self.assertEqual("cola DESC NULLS FIRST", F.col("cola").desc_nulls_first().sql())
|
||||
|
|
|
@ -335,18 +335,18 @@ class TestFunctions(unittest.TestCase):
|
|||
def test_asc_nulls_first(self):
|
||||
col_str = SF.asc_nulls_first("cola")
|
||||
self.assertIsInstance(col_str.expression, exp.Ordered)
|
||||
self.assertEqual("cola", col_str.sql())
|
||||
self.assertEqual("cola ASC", col_str.sql())
|
||||
col = SF.asc_nulls_first(SF.col("cola"))
|
||||
self.assertIsInstance(col.expression, exp.Ordered)
|
||||
self.assertEqual("cola", col.sql())
|
||||
self.assertEqual("cola ASC", col.sql())
|
||||
|
||||
def test_asc_nulls_last(self):
|
||||
col_str = SF.asc_nulls_last("cola")
|
||||
self.assertIsInstance(col_str.expression, exp.Ordered)
|
||||
self.assertEqual("cola NULLS LAST", col_str.sql())
|
||||
self.assertEqual("cola ASC NULLS LAST", col_str.sql())
|
||||
col = SF.asc_nulls_last(SF.col("cola"))
|
||||
self.assertIsInstance(col.expression, exp.Ordered)
|
||||
self.assertEqual("cola NULLS LAST", col.sql())
|
||||
self.assertEqual("cola ASC NULLS LAST", col.sql())
|
||||
|
||||
def test_desc_nulls_first(self):
|
||||
col_str = SF.desc_nulls_first("cola")
|
||||
|
|
|
@ -79,3 +79,9 @@ class TestSessionCaseSensitivity(DataFrameTestBase):
|
|||
df.sql()
|
||||
else:
|
||||
self.compare_sql(df, expected)
|
||||
|
||||
def test_alias(self):
|
||||
col = F.col('"Name"')
|
||||
self.assertEqual(col.sql(dialect=self.spark.dialect), '"Name"')
|
||||
self.assertEqual(col.alias("nAME").sql(dialect=self.spark.dialect), '"Name" AS NAME')
|
||||
self.assertEqual(col.alias('"nAME"').sql(dialect=self.spark.dialect), '"Name" AS "nAME"')
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue