generate_csv.py

import random
import sys

# 10M values to be generated
count = 10 * 1024 * 1024

def gen_repeat_in_small_range():
  for i in range(0, int(count/256/256)):
    for j in range(0, 256):
      for k in range(0, 256):
        print(j)

def gen_seq():
  for i in range(0, count):
      print(i)

def gen_seq_in_small_range():
  for i in range(0, int(count/256)):
    for j in range(0, 256):
      print(j)

def gen_random():
  for i in range(0, count):
    print(random.randint(0, count))

def gen_random_in_small_range():
  for i in range(0, count):
    print(random.randint(0, 256))


sel = int(sys.argv[1])
if sel == 1:
  gen_repeat_in_small_range()
elif sel == 2:
  gen_seq_in_small_range()
elif sel == 3:
  gen_random()
elif sel == 4:
  gen_random_in_small_range()
elif sel == 5:
  gen_seq()
else:
  raise "Invalid option"