Added a Python test.

apache · Feb 5, 2015 · c55f5fa · c55f5fa
1 parent f4b8dbb
commit c55f5fa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/pyspark/sql.py b/python/pyspark/sql.py
@@ -2638,6 +2638,9 @@ def countDistinct(col, *cols):
         >>> from pyspark.sql import Dsl
         >>> df.agg(Dsl.countDistinct(df.age, df.name).alias('c')).collect()
         [Row(c=2)]
+
+        >>> df.agg(Dsl.countDistinct("age", "name").alias('c')).collect()
+        [Row(c=2)]
         """
         sc = SparkContext._active_spark_context
         jcols = ListConverter().convert([_to_java_column(c) for c in cols],