Example #1
0
    def test_update(self):
        l1 = random.sample(range(10000), 2000)
        l2 = random.sample(range(10000), 2000)

        s2 = set(l1)
        s2.update(l2)

        s1 = IntSet(l1)
        s1.update(l2)
        self.assertTrue(list(s1) == sorted(list(s2)))

        s1 = IntSet(l1)
        s1.update(set(l2))
        self.assertTrue(list(s1) == sorted(list(s2)))

        s1 = IntSet(l1)
        s1.update(IntSet(l2))
        self.assertTrue(list(s1) == sorted(list(s2)))
        self.assertRaises(TypeError, s1.update, 1)
        self.assertRaises(TypeError, s1.update, ["1"])

        s1 = IntSet([1 << 1000])
        s1.update(range(255, 257))
        self.assertEqual(s1.max(), 1 << 1000)

        s1 = IntSet(range(10))
        s1.update(range(1 << 100, (1 << 100) + 5))
        self.assertEqual(s1.max(), (1 << 100) + 4)
Example #2
0
 def test_max(self):
     l1 = random.sample(range(10000), 2000)
     s = IntSet(l1)
     self.assertEqual(s.max(), max(l1))
     self.assertRaises(ValueError, lambda: IntSet().max())