def test_roc_notgold(self): golds = {0: -1, 1: 0, 2: 1} scores = dict([(i, Score(i, g, 0)) for i, g in golds.items()]) se = ScoreExport(scores, False) print(list(se.roc())) assert len(list(se.roc())) == 2
def test_roc(self): golds = {1: 0, 2: 0, 3: 1, 4: 1, 5: 0, 6: 1} scores = dict([(i, Score(i, g, 0)) for i, g in golds.items()]) se = ScoreExport(scores, False) print(list(se.roc())) for i, item in enumerate(se.roc()): g, p = item assert g == golds[i + 1] assert p == 0