#!/usr/bin/env python

from __future__ import print_function
import os
import subprocess

executable = ["java","-jar","../../dist/simpledb.jar"]
output_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), "output")
catalog   = "../../catalog.txt"
queries   = "queries.txt"

def run_small_experiment():
    this_output_dir = os.path.join(output_dir, "small")

    iters     = 3
    min_alpha = 0.01
    max_alpha = 0.99
    step      = 0.32

    run_experiment(this_output_dir, iters, min_alpha, max_alpha, step)

def run_medium_experiment():
    this_output_dir = os.path.join(output_dir, "medium")

    iters     = 100
    min_alpha = 0.00
    max_alpha = 1.00
    step      = 0.20

    run_experiment(this_output_dir, iters, min_alpha, max_alpha, step)

def run_experiment_large():
    this_output_dir = os.path.join(output_dir, "large")

    iters     = 500
    min_alpha = 0.00
    max_alpha = 1.00
    step      = 0.05

    run_experiment(this_output_dir, iters, min_alpha, max_alpha, step)

def run_experiment(this_output_dir, iters, min_alpha, max_alpha, step):
    if not os.path.isdir(this_output_dir):
        os.makedirs(this_output_dir)

    # Timing using ImputeDB
    subprocess.call(executable +
        ["experiment", catalog, queries, this_output_dir,
         str(iters), str(min_alpha), str(max_alpha), str(step)])

    # Timing using impute on base table
    subprocess.call(executable +
        ["experiment", catalog, queries, this_output_dir,
         str(iters), "--base"])

if __name__ == "__main__":
    import sys
    if len(sys.argv) == 2:
        experiment_size = sys.argv[1]
    else:
        experiment_size = "small"

    if experiment_size == "small":
        run_small_experiment()
    elif experiment_size == "medium":
        run_medium_experiment()
    elif experiment_size == "large":
        run_medium_experiment()
    else:
        raise Error

@@ -760,32 +760,38 @@
  step)%0A%0Adef run_
+large_
 experiment_large
@@ -784,22 +784,16 @@
 periment
-_large
 ():%0A    
@@ -1832,38 +1832,37 @@
 e%22:%0A        run_
-medium
+large
 _experiment()%0A  

Fix typo in large experiment