add multiline test

NVIDIA · Dec 6, 2021 · 90f174f · 90f174f
1 parent 1021205
commit 90f174f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/integration_tests/src/main/python/string_test.py b/integration_tests/src/main/python/string_test.py
@@ -547,6 +547,13 @@ def test_regexp_extract():
                 'regexp_extract(a, "^([a-d]*)([0-9]*)([a-d]*)$", 3)'),
             conf={'spark.rapids.sql.expression.RegExpExtract': 'true'})
 
+def test_regexp_extract_multiline():
+    gen = mk_str_gen('[abcd]{2}[\r\n]{0,2}[0-9]{2}[\r\n]{0,2}[abcd]{2}')
+    assert_gpu_and_cpu_are_equal_collect(
+            lambda spark: unary_op_df(spark, gen).selectExpr(
+                'regexp_extract(a, "^([a-d]*)([\r\n]*)", 2)'),
+            conf={'spark.rapids.sql.expression.RegExpExtract': 'true'})
+
 def test_regexp_extract_multiline_negated_character_class():
     gen = mk_str_gen('[abcd]{2}[\r\n]{0,2}[0-9]{2}[\r\n]{0,2}[abcd]{2}')
     assert_gpu_and_cpu_are_equal_collect(