fix python test

chenghao-intel · May 8, 2015 · ab35ab5 · ab35ab5
1 parent b5961a9
commit ab35ab5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pyspark/sql/dataframe.py b/python/pyspark/sql/dataframe.py
@@ -1275,7 +1275,7 @@ def __init__(self, jc):
 
     # container operators
     __contains__ = _bin_op("contains")
-    __getitem__ = _bin_op("getItem")
+    __getitem__ = getItem
 
     # bitwise operators
     bitwiseOR = _bin_op("bitwiseOR")
@@ -1300,7 +1300,7 @@ def getItem(self, key):
         |   1| value|
         +----+------+
         """
-        return self[key]
+        return Column(self._jc.getItem(key))
 
     def getField(self, name):
         """An expression that gets a field by name in a StructField.