def test_sub_treatment(self):
t = FMRITreatment(output_dir=self.tmp_dir)
t.enable_draft_testing()
sub_ts = t.split()
for sub_t in sub_ts:
sub_t.run()
def test_jde_estim_from_treatment_pck(self):
t = FMRITreatment(make_outputs=False, result_dump_file=None)
t.enable_draft_testing()
sub_ts = t.split()
sub_t_fn = op.join(self.tmp_dir, 'treatment.pck')
fout = open(sub_t_fn, 'w')
cPickle.dump(sub_ts[0], fout)
fout.close()
cmd = 'pyhrf_jde_estim -t %s' %sub_t_fn
if os.system(cmd) != 0:
raise Exception('"' + cmd + '" did not execute correctly')
