Fix test_unsupported_fallback_regexp_replace (#10093)

Signed-off-by: Haoyang Li <haoyangl@nvidia.com>
NVIDIA · Dec 29, 2023 · 9877a3f · 9877a3f
1 parent 384b2a9
commit 9877a3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integration_tests/src/main/python/regexp_test.py b/integration_tests/src/main/python/regexp_test.py
@@ -939,7 +939,7 @@ def assert_gpu_did_fallback(sql_text):
 
 @allow_non_gpu('ProjectExec', 'RegExpReplace')
 def test_unsupported_fallback_regexp_replace():
-    gen = mk_str_gen('[abcdef]{0,2}')
+    gen = StringGen('[abcdef]{0,2}').with_special_case('').with_special_pattern(r'[^\\$]{0,10}')
     regex_gen = StringGen(r'\[a-z\]\+')
     def assert_gpu_did_fallback(sql_text):
         assert_gpu_fallback_collect(lambda spark: