Remove 'six' usage from spark test

[iliakur]

What does this PR do?

This involves a small refactor of the assertion setup. Was easier to do it this way than search/replace.

Motivation

Keeping this separate because it's not like the other changes related to six.

Additional Notes

Review checklist (to be filled by reviewers)

• Feature or bugfix MUST have appropriate tests (unit, integration, e2e)
• Changelog entries must be created for modifications to shipped code
• Add the qa/skip-qa label if the PR doesn't need to be tested during QA.
• If you need to backport this PR to another branch, you can add the backport/<branch-name> label to the PR and it will automatically open a backport PR once this one is merged

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 89.95%. Comparing base (a47f5c6) to head (deafdda).
Report is 1 commits behind head on master.

Additional details and impacted files

Flag	Coverage Δ
activemq	?
cassandra	?
hive	?
hivemq	?
hudi	?
ignite	?
jboss_wildfly	?
kafka	?
presto	?
solr	?
spark	93.22% <100.00%> (-0.93%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.