def test_get_many_enough_space(self): q = Queue(10) q.put_many((1, 2, 3)) q.put_many((1, 2, 3)) self.assertEqual((1, 2), q.get_many(2)) self.assertEqual((3, 1), q.get_many(2)) self.assertEqual((2, 3), q.get_many(2))
def test_put_many_enough_space(self): q = Queue(10) q.put_many((1, 2, 3)) self.assertEqual(3, q.qsize()) self.assertEqual(1, q.get()) self.assertEqual(2, q.get()) self.assertEqual(3, q.get())
def test_put_many_too_many_items(self): q = Queue(1) msg = "items of size 3 is bigger then maxsize: 1" with self.assertRaisesRegexp(ValueError, msg): q.put_many((1, 2, 3)) q.put(None) with self.assertRaises(Full): q.put_many([1], block=False)