Skip to content

Add reusable protobuf test data helpers [databricks] [reduced-it] - #15948

Draft
thirtiseven wants to merge 1 commit into
NVIDIA:mainfrom
thirtiseven:protobuf-test-data-helpers
Draft

Add reusable protobuf test data helpers [databricks] [reduced-it]#15948
thirtiseven wants to merge 1 commit into
NVIDIA:mainfrom
thirtiseven:protobuf-test-data-helpers

Conversation

@thirtiseven

Copy link
Copy Markdown
Collaborator

Part of #14069.

Description

This PR adds reusable integration-test helpers for protobuf feature work. It centralizes the Spark-version-specific to_protobuf call, accepting either descriptor bytes or a descriptor path, and materializes deterministic protobuf payloads from a non-nullable StructGen.

Tests cover both to_protobuf API signatures, helper input validation, and Spark 3.5.2 CPU materialization using the descriptor fixture landed in #15814.

No GPU protobuf functionality is enabled by this PR.

Validation:

  • ./integration_tests/run_pyspark_from_build.sh -s -k 'protobuf_data_gen'
  • 7 tests passed on Apache Spark 3.5.2.

AI assistance: The change and PR description were prepared with Codex assistance and reviewed by the author before submission.

Checklists

Documentation

  • Updated for new or modified user-facing features or behaviors
  • No user-facing change

Testing

  • Added or modified tests to cover new code paths
  • Covered by existing tests
  • Not required

Performance

  • Tests ran and results are added in the PR description
  • Issue filed with a link in the PR description
  • Not required

Signed-off-by: Haoyang Li <haoyangl@nvidia.com>
@thirtiseven thirtiseven added test Only impacts tests data gen and removed data gen labels Sep 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

test Only impacts tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants