from typing import Any, TYPE_CHECKING

import pytest

from pyority_queue import IndexedQueue
if TYPE_CHECKING:
	from pyority_queue import Comparable


type IndexedQueueInitializer = dict[Any, Comparable] | list[tuple[Any, Comparable]] | tuple[tuple[Any, Comparable], ...]


def test_empty_creation():
	queue = IndexedQueue()
	assert len(queue) == 0


@pytest.mark.parametrize("items", (
	[],
	[("a", 0), ("b", 1), ("c", 2)],
	((0, 0), (1, 1), (2, 2)),
	((0.0, 0.0), (1.0, 1.0), (2.0, 2.0)),
	((lambda: None, 0),),
	((Exception(), 0.0),),
	(([], -1),),
	{},
	{"a": 0, "b": 1, "c": 2},
))
def test_creation(items: IndexedQueueInitializer):
	queue = IndexedQueue(items)
	assert len(queue) == len(items)


@pytest.mark.parametrize("items", (
	[],
	((0, 0),),
	((-1, -1), (3, 3)),
))
def test_iteration(items: IndexedQueueInitializer):
	queue = IndexedQueue(items)
	assert len(list(queue)) == len(items)


@pytest.mark.parametrize("items", (
	[],
	(("a", -3), ("b", 5)),
	(("c", 3.0), ("b", 2.0), ("a", 1.0)),
	(("c", 3), ("f", 6), ("h", 8), ("e", 5), ("g", 7), ("d", 4), ("b", 2), ("a", 0)),
))
def test_sorting(items: IndexedQueueInitializer):
	queue = IndexedQueue(items)
	in_order = list(queue)
	assert in_order == sorted(in_order)


def test_insertion():
	queue = IndexedQueue({"a": 1, "b": 2, "c": 3})
	queue["d"] = 4
	queue["e"] = 5
	assert list(queue) == ["a", "b", "c", "d", "e"]


def test_removal():
	queue = IndexedQueue({"a": 1, "b": 2, "c": 3})
	assert queue.pop() == "a"
	assert queue.pop() == "b"
	assert queue.pop() == "c"
	assert len(queue) == 0


def test_mixed_removal():
	queue = IndexedQueue({"a": 1, "b": 2, "c": 3})
	assert queue.pop() == "a"
	assert list(queue) == ["b", "c"]


def test_empty_removal():
	queue = IndexedQueue()
	with pytest.raises(IndexError):
		queue.pop()


def test_duplicates():
	queue = IndexedQueue[str, int]((("a", 0), ("a", 0), ("a", 2)))
	queue["b"] = 1
	queue["b"] = 3
	assert list(queue) == ["a", "b"]


def test_deletion():
	queue = IndexedQueue({"a": 1, "b": 2, "c": 3})
	del queue["a"]
	del queue["b"]
	assert list(queue) == ["c"]


def test_empty_deletion():
	queue = IndexedQueue()
	with pytest.raises(KeyError):
		del queue["a"]


def test_in():
	queue = IndexedQueue({"a": 1, "b": 2, "c": 3})
	assert "a" in queue
	assert "b" in queue
	assert "c" in queue
	assert "d" not in queue
	del queue["b"]
	assert "b" not in queue


def test_mixed_iteration():
	queue = IndexedQueue({"a": 1, "b": 2, "c": 3})
	results = []
	for char in queue:
		results.append(char)
		if len(queue):
			queue[queue.pop()] = 10
	assert results == ["a", "c", "b"]


def test_iteration_deletion():
	queue = IndexedQueue({"a": 1, "b": 2, "c": 3})
	results = []
	for char in queue:
		results.append(char)
		if "b" in queue:
			del queue["b"]
	assert results == ["a", "c"]