code: def test_prioritized_buffer_flood(
      capacity,
      wait_priority_after_sampling,
      initial_priority,
      uniform_ratio,
  ):
      buf = prioritized.PrioritizedBuffer(
          capacity=capacity,
          wait_priority_after_sampling=wait_priority_after_sampling,
      )
      for _ in range(100):
          for x in range(capacity + 1):
              if wait_priority_after_sampling:
                  buf.append(x)
              else:
                  buf.append(x, priority=initial_priority)
          for _ in range(5):
              n = random.randrange(1, capacity + 1)
              buf.sample(n, uniform_ratio=uniform_ratio)
              if wait_priority_after_sampling:
                  buf.set_last_priority([1.0] * n)
  
imports: import random
  import numpy
  import pytest
  from pfrl.collections import prioritized
label_desc: Use image node_id 2 for calling a global function with example usage: test_prioritized_buffer_flood(capacity, wait_priority_after_sampling, initial_priority, uniform_ratio) without return types