Skip to content

Commit

Permalink
Add is_spark_332cdh method to spark_session.py for integration tests.
Browse files Browse the repository at this point in the history
Signed-off-by: Rishi <[email protected]>
  • Loading branch information
sririshindra committed Oct 6, 2023
1 parent 66c2881 commit 95a9b2d
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion integration_tests/src/main/python/spark_session.py
Original file line number Diff line number Diff line change
Expand Up @@ -180,8 +180,11 @@ def is_spark_321cdh():
def is_spark_330cdh():
return "3.3.0.3.3.718" in spark_version()

def is_spark_332cdh():
return "3.3.2.3.3.719" in spark_version()

def is_spark_cdh():
return is_spark_321cdh() or is_spark_330cdh()
return is_spark_321cdh() or is_spark_330cdh() or is_spark_332cdh()

def is_databricks_version_or_later(major, minor):
spark = get_spark_i_know_what_i_am_doing()
Expand Down

0 comments on commit 95a9b2d

Please sign in to comment.