from os import sched_yield
from queue import SimpleQueue
from threading import Thread
from time import monotonic

period = 2


def _thread(queue1: SimpleQueue) -> None:
    consumed = 0
    start = monotonic()
    while queue1.get():
        consumed += 1
    end = monotonic()
    print(f'Consumed: {consumed / 1000 / (end - start):12.2f} Kitems/sec.')


def main():
    for arg in (1, 10, 100, 1000, 10_000, 100_000, 1000_000, 10_000_000):
        print(f'Testing for {arg=}')
        queue: SimpleQueue[bool] = SimpleQueue()
        thread = Thread(target=_thread, args=(queue,), daemon=True)
        thread.start()
        cnt = 0
        deadline = monotonic() + period
        while monotonic() < deadline:
            for i in range(arg):
                cnt += 1
                if queue.qsize() <= 1048576:
                    queue.put(True)
            sched_yield()  # with commented out - no bugs for any `range()`
            # time.sleep(0) # no bugs for any `range()`
        stop = monotonic()
        queue.put(False)
        thread.join()
        print(f'Produced: {cnt / 1000 / (stop - (deadline - period)):12.2f} Kitems/sec.')


main()