def test_bucket_drain(self): """ Test Bucket can drain it's contents. """ odd_bucket = Bucket(get_odd) for item in self.stream: odd_bucket(item) odds = odd_bucket.drain_contents() self.assertEqual((1, 3), tuple(odds)) self.assertEqual(deque(), odd_bucket.contents())