Skip to content

Commit e3b6b10

Browse files
gaogaotiantianzhengruifeng
authored andcommitted
[SPARK-55671][PYTHON][TESTS] Add zstandard requirement to connect requirement
### What changes were proposed in this pull request? Add `zstandard` check for connect test requirement check. ### Why are the changes needed? `zstandard` is now a requirement for connect so we might as well check it for connect tests. ### Does this PR introduce _any_ user-facing change? No. ### How was this patch tested? CI. ### Was this patch authored or co-authored using generative AI tooling? No. Closes #54464 from gaogaotiantian/add-zstandard. Authored-by: Tian Gao <gaogaotiantian@hotmail.com> Signed-off-by: Ruifeng Zheng <ruifengz@apache.org>
1 parent 3ca6c0f commit e3b6b10

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

python/pyspark/testing/utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,9 @@ def have_package(name: str) -> bool:
101101
have_grpc_status = have_package("grpc_status")
102102
grpc_status_requirement_message = "" if have_grpc_status else "No module named 'grpc_status'"
103103

104+
have_zstandard = have_package("zstandard")
105+
zstandard_requirement_message = "" if have_zstandard else "No module named 'zstandard'"
106+
104107

105108
googleapis_common_protos_requirement_message = ""
106109

@@ -140,6 +143,7 @@ def have_package(name: str) -> bool:
140143
or grpc_requirement_message
141144
or googleapis_common_protos_requirement_message
142145
or grpc_status_requirement_message
146+
or zstandard_requirement_message
143147
)
144148

145149
should_test_connect = not connect_requirement_message

0 commit comments

Comments
 (0)