Discard returned udfs when not used in tests

zero323 · zero323 · commit bce80c413832 · 2017-05-03T14:45:56.000+02:00
diff --git a/python/pyspark/sql/catalog.py b/python/pyspark/sql/catalog.py
@@ -247,12 +247,12 @@ def registerFunction(self, name, f, returnType=StringType()):
         [Row(stringLengthString(text)=u'3')]
 
         >>> from pyspark.sql.types import IntegerType
-        >>> strlen = spark.catalog.registerFunction("stringLengthInt", len, IntegerType())
+        >>> _ = spark.catalog.registerFunction("stringLengthInt", len, IntegerType())
         >>> spark.sql("SELECT stringLengthInt('test')").collect()
         [Row(stringLengthInt(test)=4)]
 
         >>> from pyspark.sql.types import IntegerType
-        >>> strlen = spark.udf.register("stringLengthInt", len, IntegerType())
+        >>> _ = spark.udf.register("stringLengthInt", len, IntegerType())
         >>> spark.sql("SELECT stringLengthInt('test')").collect()
         [Row(stringLengthInt(test)=4)]
         """
diff --git a/python/pyspark/sql/context.py b/python/pyspark/sql/context.py
@@ -195,12 +195,12 @@ def registerFunction(self, name, f, returnType=StringType()):
         [Row(stringLengthString(text)=u'3')]
 
         >>> from pyspark.sql.types import IntegerType
-        >>> strlen = sqlContext.registerFunction("stringLengthInt", lambda x: len(x), IntegerType())
+        >>> _ = sqlContext.registerFunction("stringLengthInt", lambda x: len(x), IntegerType())
         >>> sqlContext.sql("SELECT stringLengthInt('test')").collect()
         [Row(stringLengthInt(test)=4)]
 
         >>> from pyspark.sql.types import IntegerType
-        >>> strlen = sqlContext.udf.register("stringLengthInt", lambda x: len(x), IntegerType())
+        >>> _ = sqlContext.udf.register("stringLengthInt", lambda x: len(x), IntegerType())
         >>> sqlContext.sql("SELECT stringLengthInt('test')").collect()
         [Row(stringLengthInt(test)=4)]
         """
