from __future__ import absolute_import, print_function

import unittest

import silenttestrunner

class testlrucachedict(unittest.TestCase):

    def testsimple(self):

        d = util.lrucachedict(4)

        self.assertEqual(d.capacity, 4)

        d.insert('a', 'va', cost=2)

        d['b'] = 'vb'

        d['c'] = 'vc'

        d.insert('d', 'vd', cost=42)

        self.assertEqual(d['a'], 'va')

        self.assertEqual(d['b'], 'vb')

        self.assertEqual(d['c'], 'vc')

        self.assertEqual(d['d'], 'vd')

        self.assertEqual(d.totalcost, 44)

        # 'a' should be dropped because it was least recently used.

        d['e'] = 've'

        self.assertNotIn('a', d)

        self.assertIsNone(d.get('a'))

        self.assertEqual(d.totalcost, 42)

        self.assertEqual(d['b'], 'vb')

        self.assertEqual(d['c'], 'vc')

        self.assertEqual(d['d'], 'vd')

        self.assertEqual(d['e'], 've')

        # Replacing item with different cost adjusts totalcost.

        d.insert('e', 've', cost=4)

        self.assertEqual(d.totalcost, 46)

        # Touch entries in some order (both get and set).

        d['e']

        d['c'] = 'vc2'

        d['d']

        d['b'] = 'vb2'

        # 'e' should be dropped now

        d['f'] = 'vf'

        self.assertNotIn('e', d)

        self.assertEqual(d['b'], 'vb2')

        self.assertEqual(d['c'], 'vc2')

        self.assertEqual(d['d'], 'vd')

        self.assertEqual(d['f'], 'vf')

        d.clear()