Fix test

apache · Oct 10, 2014 · da14879 · da14879
1 parent 6343bcb
commit da14879
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pyspark/tests.py b/python/pyspark/tests.py
@@ -683,7 +683,7 @@ def test_udf2(self):
         self.sqlCtx.registerFunction("strlen", lambda string: len(string))
         self.sqlCtx.inferSchema(self.sc.parallelize([Row(a="test")])).registerTempTable("test")
         [res] = self.sqlCtx.sql("SELECT strlen(a) FROM test WHERE strlen(a) > 1").collect()
-        self.assertEqual(4, res[0])
+        self.assertEqual(u"4", res[0])
 
     def test_broadcast_in_udf(self):
         bar = {"a": "aa", "b": "bb", "c": "abc"}