def setUp(self):
self.fm = fixmat.TestFixmatFactory(categories = [1,2,3],
filenumbers = [1,2,3,4,5,6],
subjectindices = [1, 2, 3, 4, 5, 6],
params = {'pixels_per_degree':1, 'image_size':[100,500]})
measures.set_scores([measures.roc_model,
measures.kldiv_model,
measures.nss_model])
def test_prediction_scores(self):
measures.set_scores([measures.roc_model,
measures.kldiv_model,
measures.nss_model])
fm = fixmat.TestFixmatFactory(categories = [1,2,3],
filenumbers = [1,2,3,4,5,6],
subjectindices = [1, 2, 3, 4, 5, 6],
params = {'pixels_per_degree':10, 'image_size':[100,500]})
fdm = fixmat.compute_fdm(fm)
measures.set_scores([measures.roc_model,
measures.kldiv_model,
measures.nss_model])
scores = measures.prediction_scores(fdm, fm)
self.assertEquals(len(scores), 3)
measures.set_scores([measures.roc_model])
scores = measures.prediction_scores(fdm, fm)
self.assertEquals(len(scores), 1)
measures.set_scores([measures.roc_model,
measures.kldiv_model,
measures.nss_model])
scores = measures.prediction_scores(fdm, fm)
self.assertEquals(len(scores), 3)
