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')