From aa9469d02e651e0f0d925877e32532869bfba067 Mon Sep 17 00:00:00 2001
From: Yu ISHIKAWA <yuu.ishikawa@gmail.com>
Date: Sat, 13 Jun 2015 01:59:25 +0900
Subject: [PATCH] Fix a python test suite error caused by python 3.x

---
 python/pyspark/ml/clustering.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/python/pyspark/ml/clustering.py b/python/pyspark/ml/clustering.py
index cc0dcc7b89430..b23f7b3c2d280 100644
--- a/python/pyspark/ml/clustering.py
+++ b/python/pyspark/ml/clustering.py
@@ -49,8 +49,10 @@ class KMeans(JavaEstimator, HasFeaturesCol, HasMaxIter, HasSeed):
     >>> len(centers)
     2
     >>> transformed = model.transform(df)
-    >>> transformed.columns
-    [u'features', u'prediction']
+    >>> (transformed.columns)[0] == 'features'
+    True
+    >>> (transformed.columns)[1] == 'prediction'
+    True
     >>> rows = sorted(transformed.collect(), key = lambda r: r[0])
     >>> rows[0].prediction == rows[1].prediction
     True