0001
0002
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017
0018 from pyspark.testing.sqlutils import ReusedSQLTestCase
0019
0020
0021 class ConfTests(ReusedSQLTestCase):
0022
0023 def test_conf(self):
0024 spark = self.spark
0025 spark.conf.set("bogo", "sipeo")
0026 self.assertEqual(spark.conf.get("bogo"), "sipeo")
0027 spark.conf.set("bogo", "ta")
0028 self.assertEqual(spark.conf.get("bogo"), "ta")
0029 self.assertEqual(spark.conf.get("bogo", "not.read"), "ta")
0030 self.assertEqual(spark.conf.get("not.set", "ta"), "ta")
0031 self.assertRaisesRegexp(Exception, "not.set", lambda: spark.conf.get("not.set"))
0032 spark.conf.unset("bogo")
0033 self.assertEqual(spark.conf.get("bogo", "colombia"), "colombia")
0034
0035 self.assertEqual(spark.conf.get("hyukjin", None), None)
0036
0037
0038
0039
0040 self.assertEqual(spark.conf.get("spark.sql.sources.partitionOverwriteMode"), "STATIC")
0041
0042
0043
0044 self.assertEqual(spark.conf.get("spark.sql.sources.partitionOverwriteMode", None), None)
0045
0046
0047 if __name__ == "__main__":
0048 import unittest
0049 from pyspark.sql.tests.test_conf import *
0050
0051 try:
0052 import xmlrunner
0053 testRunner = xmlrunner.XMLTestRunner(output='target/test-reports', verbosity=2)
0054 except ImportError:
0055 testRunner = None
0056 unittest.main(testRunner=testRunner, verbosity=2)