def test_task_entry_collections(self):
# setup
subj = models.Subject(name="test_subject")
subj.save()
subj = models.Subject.objects.get(name="test_subject")
task = models.Task(name="test_task")
task.save()
task = models.Task.objects.get(name="test_task")
te1 = models.TaskEntry(subject_id=subj.id, task_id=task.id)
te1.save()
te2 = models.TaskEntry(subject_id=subj.id, task_id=task.id)
te2.save()
col = models.TaskEntryCollection(name="new_col")
col.save()
col.add_entry(te1)
self.assertEqual(len(col.entries.all()), 1)
# adding the same entry twice shouldn't do anything
col.add_entry(te1)
self.assertEqual(len(col.entries.all()), 1)
# adding a second entry should increase the length of the list
col.add_entry(te2)
self.assertEqual(len(col.entries.all()), 2)
# remove_entry should cause a change in the list
col.remove_entry(te1)
self.assertEqual(len(col.entries.all()), 1)
self.assertEqual(col.entries.all()[0].id, te2.id)
def test_start_experiment_python(self):
import json
from built_in_tasks.passivetasks import TargetCaptureVFB2DWindow
from riglib import experiment
from features import Autostart
from tracker import json_param
try:
import pygame
except ImportError:
print("Skipping test due to pygame missing")
return
# Create all the needed database entries
subj = models.Subject(name="test_subject")
subj.save()
task = models.Task(
name="test_vfb",
import_path="built_in_tasks.passivetasks.TargetCaptureVFB2DWindow")
task.save()
models.Generator.populate()
gen = models.Generator.objects.get(name='centerout_2D_discrete')
seq_params = dict(nblocks=1, ntargets=1)
seq_rec = models.Sequence(generator=gen,
params=json.dumps(seq_params),
task=task)
seq_rec.save()
print(seq_rec)
task_rec = models.Task.objects.get(name='test_vfb')
te = models.TaskEntry(task=task_rec, subject=subj)
te.save()
seq, seq_params = seq_rec.get()
# Start the task
base_class = task.get_base_class()
Task = experiment.make(base_class, feats=[])
params = json_param.Parameters.from_dict(dict(window_size=(480, 240)))
params.trait_norm(Task.class_traits())
saveid = te.id
task_start_data = dict(subj=subj.id,
base_class=base_class,
feats=[Autostart],
params=dict(window_size=(480, 240)),
seq=seq_rec,
seq_params=seq_params,
saveid=saveid)
tracker = tasktrack.Track.get_instance()
tracker.runtask(cli=True, **task_start_data)
def test_list_exp_history(self):
subj = models.Subject(name="test_subject")
subj.save()
subj = models.Subject.objects.get(name="test_subject")
task = models.Task(name="test_task")
task.save()
task = models.Task.objects.get(name="test_task")
list_data0 = views._list_exp_history()
self.assertEqual(len(list_data0['entries']), 0)
self.assertEqual(len(list_data0['subjects']), 1)
self.assertEqual(len(list_data0['tasks']), 1)
self.assertEqual(len(list_data0['features']), 0)
self.assertEqual(len(list_data0['generators']), 0)
te1 = models.TaskEntry(subject_id=subj.id,
task_id=task.id,
date=datetime.datetime.now())
te1.save()
list_data1 = views._list_exp_history()
self.assertEqual(len(list_data1['entries']), 1)
self.assertEqual(len(list_data1['subjects']), 1)
self.assertEqual(len(list_data1['tasks']), 1)
self.assertEqual(len(list_data1['features']), 0)
self.assertEqual(len(list_data1['generators']), 0)
for k in range(300):
te2_date = datetime.datetime.now() - datetime.timedelta(days=k)
te2 = models.TaskEntry(subject_id=subj.id,
task_id=task.id,
date=te2_date)
te2.save()
# all entries returned if no args
list_data2 = views._list_exp_history()
self.assertEqual(len(list_data2['entries']), 301)
# 'listall' should return all entries
list_data3 = views._list_exp_history(max_entries=100)
self.assertEqual(len(list_data3['entries']), 100)
def test_create_task_entry(self):
"""
"""
subj = models.Subject(name="test_subject")
subj.save()
subj = models.Subject.objects.get(name="test_subject")
task = models.Task(name="test_task")
task.save()
task = models.Task.objects.get(name="test_task")
te = models.TaskEntry(subject_id=subj.id, task_id=task.id)
te.save()
def setUp(self):
subj = models.Subject(name="test_subject")
subj.save()
subj = models.Subject.objects.get(name="test_subject")
task = models.Task(name="test_task")
task.save()
task = models.Task.objects.get(name="test_task")
te = models.TaskEntry(subject_id=subj.id, task_id=task.id)
te.save()
system = models.System(name="test_system", path="", archive="")
system.save()
