test(spark): Improve `test_spark` speed #4160

Job	Run time
Cloud (pinned) (3.12, ubuntu-22.04)	2m 50s
Cloud (pinned) (3.11, ubuntu-22.04)	3m 37s
Cloud (pinned) (3.9, ubuntu-22.04)	4m 46s
Cloud (pinned) (3.6, ubuntu-22.04)	3m 23s
Cloud (pinned) (3.10, ubuntu-22.04)	1m 42s
Cloud (pinned) (3.8, ubuntu-22.04)	4m 41s
Cloud (pinned) (3.7, ubuntu-22.04)	4m 4s
Cloud (pinned) (3.13, ubuntu-22.04)	5m 53s
All pinned Cloud tests passed	2s
	30m 58s

Provide feedback