def test_get_closest(self): cache = PnrpCache(1000, 1, 0, 1) self.assertEquals(1, cache.get_closest(0)) cache.add(50, 2) self.assertEquals(1, cache.get_closest(4)) self.assertEquals(2, cache.get_closest(26)) self.assertEquals(2, cache.get_closest(400)) self.assertEquals(1, cache.get_closest(600)) self.assertEquals(1, cache.get_closest(997)) cache.add(500, 3) self.assertEquals(2, cache.get_closest(200)) self.assertEquals(3, cache.get_closest(300)) self.assertEquals(3, cache.get_closest(740)) self.assertEquals(1, cache.get_closest(760))