def test_decision_level(self): r_level = 3 package = self.mkl_10_1_0 decisions = DecisionsSet(self.pool) decisions.decide(package.id, r_level, "because") self.assertEqual(decisions.decision_level(package.id), r_level) self.assertEqual(decisions.decision_level(-package.id), r_level) self.assertEqual(decisions.decision_level(package.id + 1), 0)