changed DataStore.generator() to DataStore.load_chunks() and modified tests