Skip to content

[FEATURE]: Migrate to pyspark testing utils #1105

@mwojtyczka

Description

@mwojtyczka

Is there an existing issue for this?

  • I have searched the existing issues

Problem statement

DQX uses chispa (assert_df_equality) as a test dependency. Migrating to pyspark testing utils (available from pyspark 3.5) would remove the 3rd party dependency.

Differences:

  • chispa has ignore_nullable=True and ignore_column_order=True which are used extensively. assertDataFrameEqual supports these but with slightly different parameter names (check_nullable, check_column_order — need to verify the exact API).

Proposed Solution

The migration would be:

  1. Replace chispa import with from pyspark.testing.utils import assertDataFrameEqual
  2. Update assert_df_equality_ignore_fingerprints to use assertDataFrameEqual internally
  3. Map parameter names (ignore_nullable → PySpark equivalent)
  4. Remove chispa from pyproject.toml

It's straightforward but touches every test file.

Additional Context

No response

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions