def test_generate(self):
# Test loading experiment folder path
generate(self.battery)
self.assertTrue(os.path.exists(self.battery))
# Test loading experiments
experiments = get_experiments("%s/static/experiments" %self.battery,load=True)
self.assertTrue(len(experiments)>0)
def test_generate(self):
# Test loading experiment folder path
generate(self.battery)
self.assertTrue(os.path.exists(self.battery))
# Test loading experiments
experiments = get_experiments("%s/static/experiments" % self.battery,
load=True)
self.assertTrue(len(experiments) > 0)
def select():
if request.method == 'POST':
fields = dict()
for field, value in request.form.items():
if field == "deploychoice":
deploychoice = value
else:
fields[field] = value
# Retrieve experiment folders
valid_experiments = app.experiments
experiments = [x[0]["exp_id"] for x in valid_experiments]
selected_experiments = [x for x in fields.values() if x in experiments]
experiment_folders = [
"%s/experiments/%s" % (app.tmpdir, x) for x in selected_experiments
]
# Option 1: A folder on the local machine
if deploychoice == "folder":
# Add to the battery
generate(battery_dest="%s/expfactory-battery" % app.tmpdir,
battery_repo="%s/battery" % app.tmpdir,
experiment_repo="%s/experiments" % app.tmpdir,
experiments=experiment_folders,
make_config=False,
warning=False)
battery_dest = "%s/expfactory-battery" % (app.tmpdir)
# Option 2 or 3: Virtual machine (vagrant) or cloud (aws)
else:
specify_experiments(battery_dest=app.tmpdir,
experiments=selected_experiments)
battery_dest = app.tmpdir
# Clean up
clean_up("%s/experiments" % (app.tmpdir))
clean_up("%s/battery" % (app.tmpdir))
clean_up("%s/vm" % (app.tmpdir))
return render_template('complete.html', battery_dest=battery_dest)
def select():
if request.method == 'POST':
fields = dict()
for field,value in request.form.iteritems():
if field == "deploychoice":
deploychoice = value
else:
fields[field] = value
# Retrieve experiment folders
valid_experiments = app.experiments
experiments = [x[0]["exp_id"] for x in valid_experiments]
selected_experiments = [x for x in fields.values() if x in experiments]
experiment_folders = ["%s/experiments/%s" %(app.tmpdir,x) for x in selected_experiments]
# Option 1: A folder on the local machine
if deploychoice == "folder":
# Add to the battery
generate(battery_dest="%s/expfactory-battery"%app.tmpdir,
battery_repo="%s/battery"%app.tmpdir,
experiment_repo="%s/experiments"%app.tmpdir,
experiments=experiment_folders,
make_config=False,
warning=False)
battery_dest = "%s/expfactory-battery" %(app.tmpdir)
# Option 2 or 3: Virtual machine (vagrant) or cloud (aws)
else:
specify_experiments(battery_dest=app.tmpdir,experiments=selected_experiments)
battery_dest = app.tmpdir
# Clean up
clean_up("%s/experiments"%(app.tmpdir))
clean_up("%s/battery"%(app.tmpdir))
clean_up("%s/vm"%(app.tmpdir))
return render_template('complete.html',battery_dest=battery_dest)
#!/usr/bin/python from expfactory.battery import generate import sys # Standard battery setup script # This script will generate a battery for all valid experiments, taking as input # a battery folder, an experiment folder, and an output directory. battery_dest = sys.argv[1] generate(battery_dest=battery_dest)
#!/usr/bin/python from expfactory.battery import generate # Location of battery repo, and experiment repo # In future, these will not be required battery_repo = "/home/vanessa/Documents/Dropbox/Code/psiturk/psiturk-battery" experiment_repo = "/home/vanessa/Documents/Dropbox/Code/psiturk/psiturk-experiments" battery_dest = "/home/vanessa/Desktop/battery" ### This is the command line way to generate a battery # config parameters are specified via dictionary # Not specifying experiments will include all valid generate(battery_repo,battery_dest,experiment_repo)
def main():
parser = argparse.ArgumentParser(
description="generate experiments and infrastructure to serve them.")
parser.add_argument("--folder", dest='folder', help="full path to single experiment folder (for single experiment run with --run) or folder with many experiments (for battery run with --run)", type=str, default=None)
parser.add_argument("--subid", dest='subid', help="subject id to embed in experiments data in the case of a battery run with --run", type=str, default=None)
parser.add_argument("--experiments", dest='experiments', help="comma separated list of experiments for a local battery", type=str, default=None)
parser.add_argument("--port", dest='port', help="port to preview experiment", type=int, default=None)
parser.add_argument("--battery", dest='battery_folder', help="full path to local battery folder to use as template", type=str, default=None)
parser.add_argument("--time", dest='time', help="maximum number of minutes for battery to endure, to select experiments", type=int, default=99999)
parser.add_argument('--preview', help="preview an experiment locally (development function)", dest='preview', default=False, action='store_true')
parser.add_argument('--run', help="run a single experiment/survey or battery locally", dest='run', default=False, action='store_true')
parser.add_argument("--survey", dest='survey', help="survey to run for a local assessment", type=str, default=None)
parser.add_argument("--game", dest='game', help="game to run for a local assessment", type=str, default=None)
parser.add_argument('--validate', dest='validate', help="validate an experiment folder", default=False, action='store_true')
parser.add_argument('--psiturk', dest='psiturk', help="to be used with the --generate command, to generate a psiturk battery instead of local folder deployment", default=False, action='store_true')
parser.add_argument('--generate', dest='generate', help="generate (and don't run) a battery with --experiments to a --folder", default=False, action='store_true')
parser.add_argument("--output", dest='output', help="output folder for --generate command, if a temporary directory is not desired. Must not exist.", type=str, default=None)
parser.add_argument('--test', dest='test', help="test an experiment folder with the experiment robot", default=False, action='store_true')
try:
args = parser.parse_args()
except:
parser.print_help()
sys.exit(0)
# Check if the person wants to preview experiment or battery
if args.preview == True:
preview_experiment(folder=args.folder,battery_folder=args.battery_folder,port=args.port)
# Generate a local battery folder (static)
elif args.generate == True:
if args.experiments != None:
# Deploy a psiturk battery folder
experiments = args.experiments.split(",")
if args.psiturk == True:
outdir = generate(battery_dest=args.output,
battery_repo=args.battery_folder,
experiment_repo=args.folder,
experiments=experiments,
make_config=True,
warning=False)
# Deploy a regular battery folder
else:
outdir = generate_local(battery_dest=args.output,
subject_id="expfactory_battery_result",
battery_repo=args.battery_folder,
experiment_repo=args.folder,
experiments=experiments,
warning=False,
time=args.time)
print "Battery generation complete: static files are in %s" %(outdir)
else:
print "Please specify list of comma separated experiments with --experiments"
# Run a local battery
elif args.run == True:
# Warn the user about using repos for experiments and battery
if args.battery_folder == None:
print "No battery folder specified. Will pull latest battery from expfactory-battery repo"
if args.folder == None:
print "No experiments, games, or surveys folder specified. Will pull latest from expfactory-experiments repo"
if args.survey != None:
survey = args.survey.split(",")
if len(survey) > 0:
print "Currently only support running one survey, will run first in list."
survey = survey[0]
run_single(exp_id=survey,
repo_type="surveys",
source_repo=args.folder,
battery_repo=args.battery_folder,
port=args.port,
subject_id=args.subid)
if args.game != None:
game = args.game.split(",")
if len(game) > 0:
print "Currently only support running one game, will run first in list."
game = game[0]
run_single(exp_id=game,
repo_type="games",
source_repo=args.folder,
battery_repo=args.battery_folder,
port=args.port,
subject_id=args.subid)
if args.experiments != None:
experiments = args.experiments.split(",")
run_battery(experiments=experiments,
experiment_folder=args.folder,
subject_id=args.subid,
battery_folder=args.battery_folder,
port=args.port,
time=args.time)
else:
print "Please specify list of comma separated experiments with --experiments"
# Validate a config.json
elif args.validate == True:
if args.folder == None:
folder = os.getcwd()
validate(experiment_folder=folder)
# If a survey, and if validates, also validate survey.tsv
experiment = load_experiment(folder)[0]
if experiment["template"] == "survey":
print "Validating survey.tsv..."
survey_repo = os.path.dirname(folder)
validate_surveys(experiment["exp_id"],survey_repo)
# Run the experiment robot
elif args.test == True:
from expfactory.tests import test_experiment
test_experiment(folder=args.folder,battery_folder=args.battery_folder,port=args.port)
# Otherwise, just open the expfactory interface
else:
from expfactory.interface import start
start(port=args.port)
