Uploaded image for project: 'Spark'
  1. Spark
  2. SPARK-27921

Convert applicable *.sql tests into UDF integrated test base

    XMLWordPrintableJSON

Details

    • Umbrella
    • Status: Resolved
    • Major
    • Resolution: Done
    • 3.0.0
    • 3.0.0
    • PySpark, SQL, Tests
    • None

    Description

      This JIRA targets to improve Python test coverage in particular about ExtractPythonUDFs.
      This rule has caused many regressions or issues such as SPARK-27803, SPARK-26147, SPARK-26864, SPARK-26293, SPARK-25314 and SPARK-24721.
      We should convert *.sql test cases that can be affected by this rule ExtractPythonUDFs like https://github.com/apache/spark/blob/f5317f10b25bd193cf5026a8f4fd1cd1ded8f5b4/sql/core/src/test/resources/sql-tests/inputs/udf/udf-inner-join.sql
      Namely most of plan related test cases might have to be converted.

      Here is the rough contribution guide to follow:

      Make sure you have Python with Pandas 0.23.2+ and PyArrow 0.12.1+. Check if you're able to do this:

      >>> import pandas
      >>> pandas.__version__
      '0.23.4'
      >>> import pyarrow
      >>> pyarrow.__version__
      '0.13.0'
      >>> pyarrow.Table.from_pandas(pandas.DataFrame({'a': [1,2,3]}))
      pyarrow.Table
      a: int64
      metadata
      --------
      OrderedDict([(b'pandas',
                    b'{"index_columns": [{"kind": "range", "name": null, "start": '
                    b'0, "stop": 3, "step": 1}], "column_indexes": [{"name": null,'
                    b' "field_name": null, "pandas_type": "unicode", "numpy_type":'
                    b' "object", "metadata": {"encoding": "UTF-8"}}], "columns": ['
                    b'{"name": "a", "field_name": "a", "pandas_type": "int64", "nu'
                    b'mpy_type": "int64", "metadata": null}], "creator": {"library'
                    b'": "pyarrow", "version": "0.13.0"}, "pandas_version": null}')])
      

       
      1. Copy and paste sql/core/src/test/resources/sql-tests/inputs/xxx.sql file into sql/core/src/test/resources/sql-tests/inputs/udf/udf-xxx.sql

      2. Keep the comments and state that this file was copied from sql/core/src/test/resources/sql-tests/inputs/xxx.sql, for now.
      For instance, let's add a comment as below on the top:

      -- This test file was converted from xxx.sql.
      

      3. Run it below:

      SPARK_GENERATE_GOLDEN_FILES=1 build/sbt "sql/test-only *SQLQueryTestSuite -- -z udf/udf-xxx.sql"
      git add .
      

      4. Insert one or multiple udf(...) into each statement. It is not required to add more combinations.
      And it is not strict about where to insert. Ideally, we should try to put udf differently for each statement.

      5. Run it below again:

      SPARK_GENERATE_GOLDEN_FILES=1 build/sbt "sql/test-only *SQLQueryTestSuite -- -z udf/udf-xxx.sql"
      git diff
      # or git diff --no-index sql/core/src/test/resources/sql-tests/results/xxx.sql.out sql/core/src/test/resources/sql-tests/results/udf/xxx.sql.out
      

      6. Compare results with original file, sql/core/src/test/resources/sql-tests/results/xxx.sql.out

      7. If there are diff, analyze it, file or find the JIRA, skip the tests with comments. Please see this comment when you file a JIRA.
      It's more than perfect if you are even able to fix an issue found but this can be done separately. There is a great example to check and follow at SPARK-28323, done by viirya

      8. Run without generating golden files and check:

      build/sbt "sql/test-only *SQLQueryTestSuite -- -z udf/udf-xxx.sql"
      

      9. When you open a PR. please attach git diff --no-index sql/core/src/test/resources/sql-tests/results/xxx.sql.out sql/core/src/test/resources/sql-tests/results/udf/xxx.sql.out in the PR description with the template below:

      <details><summary>Diff comparing to 'xxx.sql'</summary>
      <p>
      
      ```diff
      ...  # here you put 'git diff' results
      ```
      
      </p>
      </details>
      

      10. You're ready. Please go for a PR! If the PR contains other minor fixes, use [SPARK-XXXXX][SQL][PYTHON] prefix in the PR title. If the PR is purely about tests, use [SPARK-XXXXX][SQL][PYTHON][TESTS].
      See https://github.com/apache/spark/pull/25069 as an example.

      Note that registered UDFs all return strings - so there are some differences are expected.
      Note that this JIRA targets plan specific cases in general.
      Note that one output.sql.out file is shared for three UDF test cases (Scala UDF, Python UDF, and Pandas UDF). Beware of it when you fix the tests.
      Note that this guide is supposed to be updated continuously given how it goes.
      Note that this test case uses the integrated UDF test base. See https://github.com/apache/spark/pull/24752 if you're interested in it or find an issue.

      Attachments

        Issue Links

          1.
          Convert and port 'pgSQL/aggregates_part1.sql' into UDF test base Sub-task Resolved Hyukjin Kwon
          2.
          Convert and port 'pgSQL/aggregates_part2.sql' into UDF test base Sub-task Resolved Terry Kim
          3.
          Convert and port 'pgSQL/aggregates_part3.sql' into UDF test base Sub-task Resolved Hyukjin Kwon
          4.
          Convert and port 'pgSQL/aggregates_part4.sql' into UDF test base Sub-task Resolved Hyukjin Kwon
          5.
          Convert and port 'pgSQL/case.sql' into UDF test base Sub-task Resolved Hyukjin Kwon
          6.
          Convert and port 'pgSQL/window.sql' into UDF test base Sub-task Resolved Unassigned
          7.
          Convert and port 'pgSQL/select_having.sql' into UDF test base Sub-task Resolved Shivu Sondur
          8.
          Convert and port 'pgSQL/select_implicit.sql' into UDF test base Sub-task Resolved Udbhav Agrawal
          9.
          Convert and port 'pgSQL/join.sql' into UDF test base Sub-task Resolved Huaxin Gao
          10.
          Convert and port 'cross-join.sql' into UDF test base Sub-task Resolved L. C. Hsieh
          11.
          Convert and port 'natural-join.sql' into UDF test base Sub-task Resolved Manu Zhang
          12.
          Convert and port 'outer-join.sql' into UDF test base Sub-task Resolved Huaxin Gao
          13.
          Convert and port 'join-empty-relation.sql' into UDF test base Sub-task Resolved Terry Kim
          14.
          Convert and port 'except.sql' into UDF test base Sub-task Resolved Huaxin Gao
          15.
          Convert and port 'except-all.sql' into UDF test base Sub-task Resolved Terry Kim
          16.
          Convert and port 'intersect-all.sql' into UDF test base Sub-task Resolved Terry Kim
          17.
          Convert and port 'union.sql' into UDF test base Sub-task Resolved Yiheng Wang
          18.
          Convert and port 'group-analytics.sql' into UDF test base Sub-task Resolved Stavros Kontopoulos
          19.
          Convert and port 'group-by.sql' into UDF test base Sub-task Resolved Stavros Kontopoulos
          20.
          Convert and port 'having.sql' into UDF test base Sub-task Resolved Huaxin Gao
          21.
          Convert and port 'pivot.sql' into UDF test base Sub-task Resolved Chitral Verma
          22.
          Convert and port 'inline-table.sql' into UDF test base Sub-task Resolved Terry Kim
          23.
          Convert and port 'count.sql' into UDF test base Sub-task Resolved Vinod KC
          24.
          Convert and port 'udaf.sql' into UDF test base Sub-task Resolved Vinod KC
          25.
          Convert and port 'window.sql' into UDF test base Sub-task Resolved YoungGyu Chun
          26.
          Converted PostgreSQL UDF test should change some default config Sub-task Resolved Hyukjin Kwon

          Activity

            People

              gurwls223 Hyukjin Kwon
              gurwls223 Hyukjin Kwon
              Votes:
              1 Vote for this issue
              Watchers:
              6 Start watching this issue

              Dates

                Created:
                Updated:
                Resolved: