It'd also be nice to have tests for the read-data yes/no logic. e.g. Check that np.fromfile is only called to load the data when read-data is yes.