Skip to content

Commit 944f6da

Browse files
authored
chore: remove unnecessary testing logic for pyspark (#2170)
1 parent b508e63 commit 944f6da

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

tests/utils.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -87,10 +87,6 @@ def _sort_dict_by_key(
8787

8888

8989
def assert_equal_data(result: Any, expected: Mapping[str, Any]) -> None:
90-
is_pyspark = (
91-
hasattr(result, "_compliant_frame")
92-
and result.implementation is Implementation.PYSPARK
93-
)
9490
is_duckdb = (
9591
hasattr(result, "_compliant_frame")
9692
and result._compliant_frame._implementation is Implementation.DUCKDB
@@ -111,10 +107,6 @@ def assert_equal_data(result: Any, expected: Mapping[str, Any]) -> None:
111107
for idx, (col, key) in enumerate(zip(result.columns, expected.keys())):
112108
assert col == key, f"Expected column name {key} at index {idx}, found {col}"
113109
result = {key: _to_comparable_list(result[key]) for key in expected}
114-
if (is_pyspark or is_duckdb) and expected: # pragma: no cover
115-
sort_key = next(iter(expected.keys()))
116-
expected = _sort_dict_by_key(expected, sort_key)
117-
result = _sort_dict_by_key(result, sort_key)
118110
assert list(result.keys()) == list(expected.keys()), (
119111
f"Result keys {result.keys()}, expected keys: {expected.keys()}"
120112
)

0 commit comments

Comments
 (0)