From f01ecb67d4c75c82a454de5f4c9f9a23fcdb331c Mon Sep 17 00:00:00 2001 From: Yuan Cao Date: Thu, 5 Apr 2018 21:45:59 -0400 Subject: [PATCH] Add main function to allow easier importing --- shuffle.py | 21 +++++++++++++-------- 1 file changed, 13 insertions(+), 8 deletions(-) diff --git a/shuffle.py b/shuffle.py index 2244d3e..d907ba9 100644 --- a/shuffle.py +++ b/shuffle.py @@ -256,12 +256,17 @@ def twice_shuffled(num_shards, parallel_reads): shuffled_shards.append(pseudoshuffle(subshards, buffer_size=buffer_size, parallel_reads=parallel_reads)) return pseudoshuffle(shuffled_shards, buffer_size=buffer_size, parallel_reads=parallel_reads) -for i in range(1, 6): - make_hilbert_curve_svg(i, os.path.join(OUTPUT_DIR, 'hilbert_curve_{}.svg'.format(i))) +def main(): -create_img_table(basic_scaling, (1024, 4096, 16384), (0, 0.01, 0.1, 0.5, 1)) -create_img_table(chained_scaling, (0, 0.01, 0.1, 0.5), (1, 2, 4)) -create_img_table(sharded_scaling, (0, 0.01, 0.1, 0.5), (1, 2, 4, 8)) -create_img_table(parallel_read_scaling, (1, 2, 4, 8), (1, 2, 4, 8)) -create_img_table(parallel_read_scaling_jittered, (1, 2, 4, 8), (1, 2, 4, 8)) -create_img_table(twice_shuffled, (1, 2, 4, 8), (1, 2, 4, 8)) + for i in range(1, 6): + make_hilbert_curve_svg(i, os.path.join(OUTPUT_DIR, 'hilbert_curve_{}.svg'.format(i))) + + create_img_table(basic_scaling, (1024, 4096, 16384), (0, 0.01, 0.1, 0.5, 1)) + create_img_table(chained_scaling, (0, 0.01, 0.1, 0.5), (1, 2, 4)) + create_img_table(sharded_scaling, (0, 0.01, 0.1, 0.5), (1, 2, 4, 8)) + create_img_table(parallel_read_scaling, (1, 2, 4, 8), (1, 2, 4, 8)) + create_img_table(parallel_read_scaling_jittered, (1, 2, 4, 8), (1, 2, 4, 8)) + create_img_table(twice_shuffled, (1, 2, 4, 8), (1, 2, 4, 8)) + +if __name__ == "__main__": + main()