def testGetLetorDcg(self): ev = LetorNdcgEval() self.assertAlmostEqual(ev.get_dcg([0, 0, 1, 0]), 0.6309298) self.assertEqual(ev.get_dcg([0, 1, 0, 0]), 1) self.assertEqual(ev.get_dcg([0, 0, 0, 0]), 0)