Spark version of a tool similar to TestDFSIO