Refactor tests for HDFS backed RDD