Fix tests + simplify sc.getRDDStorageInfo

This adds more beef to the test that was failing.