From a28affe9af6e7507b094a526df5feab7aa6522c9 Mon Sep 17 00:00:00 2001 From: Chong Gao Date: Wed, 11 Sep 2024 14:30:18 +0800 Subject: [PATCH] Refactor test case --- integration_tests/src/main/python/date_time_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/integration_tests/src/main/python/date_time_test.py b/integration_tests/src/main/python/date_time_test.py index c2a6a604945..7d5700b7f4c 100644 --- a/integration_tests/src/main/python/date_time_test.py +++ b/integration_tests/src/main/python/date_time_test.py @@ -675,10 +675,10 @@ def test_timestamp_micros(data_gen): @pytest.mark.parametrize('parser_policy', ["LEGACY", "CORRECTED", "EXCEPTION"], ids=idfn) def test_date_to_timestamp(parser_policy): - parser_policy_dic = {"spark.sql.legacy.timeParserPolicy": "{}".format(parser_policy)} - incompatible_dic = {"spark.rapids.sql.incompatibleDateFormats.enabled": True} assert_gpu_and_cpu_are_equal_sql( lambda spark : unary_op_df(spark, date_gen), "tab", "SELECT cast(a as timestamp) from tab", - conf=copy_and_update(parser_policy_dic, incompatible_dic)) + conf = { + "spark.sql.legacy.timeParserPolicy": "{}".format(parser_policy), + "spark.rapids.sql.incompatibleDateFormats.enabled": True})