It seems like this test writes a very large amount of data to disk. Spark has hundreds of tests and each one can only take a short amount of time. So it would be great if you could write a test that just writes a small amount of data to disk.