add import

Author: cloud-fan

python/pyspark/sql/tests.py

@@ -30,6 +30,13 @@
 import time
 import datetime
 
+if sys.version >= '3':
+    basestring = unicode = str
+    long = int
+    from functools import reduce
+else:
+    from itertools import imap as map
+
 import py4j
 try:
     import xmlrunner