Exemplo n.º 1
0
    def test_popitem(self):
        # dict.popitem()
        for copymode in -1, +1:
            # -1: b has same structure as a
            # +1: b is a.copy()
            for log2size in range(12):
                size = 2**log2size
                a = LazyDict()
                b = LazyDict()
                for i in range(size):
                    a[repr(i)] = i
                    if copymode < 0:
                        b[repr(i)] = i
                if copymode > 0:
                    b = a.copy()
                for i in range(size):
                    ka, va = ta = a.popitem()
                    self.assertEqual(va, int(ka))
                    kb, vb = tb = b.popitem()
                    self.assertEqual(vb, int(kb))
                    self.assertFalse(copymode < 0 and ta != tb)
                self.assertFalse(a)
                self.assertFalse(b)

        d = LazyDict()
        self.assertRaises(KeyError, d.popitem)
Exemplo n.º 2
0
 def test_lazy_popitem(self):
     d = LazyDict({1: 1})
     d.set_stub(2, lambda x:x)
     k, v = d.popitem()
     self.assertEqual(k, v)
     k, v = d.popitem()
     self.assertEqual(k, v)
     self.assertEqual(k, 2)
     self.assertEqual(len(d._stubs), 0)
     self.assertRaises(KeyError, d.popitem)