Clarify the usage of Row.

apache · Nov 13, 2017 · e919ed5 · e919ed5
1 parent 8b1a4d8
commit e919ed5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/pyspark/sql/tests.py b/python/pyspark/sql/tests.py
@@ -3265,7 +3265,9 @@ def test_createDataFrame_respect_session_timezone(self):
 
             self.assertNotEqual(result_ny, result_la)
 
-            result_la_corrected = [Row(*(r[0:6] + (r[6] - timedelta(hours=3),))) for r in result_la]
+            result_la_corrected = [Row(**{k: v - timedelta(hours=3) if k == '7_timestamp_t' else v
+                                          for k, v in row.asDict().items()})
+                                   for row in result_la]
             self.assertEqual(result_ny, result_la_corrected)
         finally:
             self.spark.conf.set("spark.sql.session.timeZone", orig_tz)