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)