# stdlib
import re
from operator import floordiv, truediv
from typing import List, Union

# 3rd party
import pytest

# this package
from domdf_python_tools.pagesizes.units import Unit, Unitcm, UnitInch, Unitmm, Unitpc, Unitpt, Unitum

units_of_12: List[Unit] = [Unit(12), UnitInch(12), Unitcm(12), Unitmm(12), Unitpc(12), Unitpt(12), Unitum(12)]
zero2thousand: List[int] = list(range(0, 1000, 10))
one2thousand: List[int] = list(range(1, 1000, 10))
units_zero2thousand: List[Unit] = [Unit(x) for x in zero2thousand]
units_ints_zero2thousand: List[Union[Unit, int]] = [*units_zero2thousand, *zero2thousand]


@pytest.mark.parametrize("unit", units_of_12)
def test_repr(unit: Unit):
	assert re.match(r"<Unit '12\.000 .*': .*pt", repr(unit))


@pytest.mark.parametrize("unit", units_of_12)
def test_str(unit: Unit):
	assert re.match(r"<Unit '12\.000\u205F.*': .*pt", str(unit))


class TestMul:

	@pytest.mark.parametrize("obj", units_zero2thousand)
	def test_mul_errors(self, obj: Unit):
		with pytest.raises(NotImplementedError, match="Multiplying a unit by another unit is not allowed."):
			Unit(17) * obj  # pylint: disable=expression-not-assigned

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_mul(self, obj: int):
		assert obj * Unit(17) == obj * 17
		assert obj * Unit(17 / 2) == obj * (17 / 2)

	@pytest.mark.parametrize("obj", units_zero2thousand)
	def test_rmul_errors(self, obj: Unit):
		with pytest.raises(NotImplementedError, match="Multiplying a unit by another unit is not allowed."):
			obj * Unit(17)  # pylint: disable=expression-not-assigned

	@pytest.mark.parametrize("obj", units_ints_zero2thousand)
	def test_pow_errors(self, obj: Unit):
		with pytest.raises(NotImplementedError, match="Powers are not supported for units."):
			Unit(17)**obj  # pylint: disable=expression-not-assigned

		if isinstance(obj, Unit):
			with pytest.raises(NotImplementedError, match="Powers are not supported for units."):
				obj**Unit(17)  # pylint: disable=expression-not-assigned


class TestDiv:

	@pytest.mark.parametrize("obj", one2thousand)
	def test_truediv(self, obj: int):
		assert truediv(Unit(17), obj) == 17 / obj
		assert truediv(Unit(17 / 2), obj) == (17 / 2) / obj

	@pytest.mark.parametrize("obj", units_zero2thousand)
	def test_truediv_errors(self, obj: Unit):
		with pytest.raises(NotImplementedError, match="Dividing a unit by another unit is not allowed."):
			truediv(Unit(17), obj)
		with pytest.raises(NotImplementedError, match="Dividing a unit by another unit is not allowed."):
			truediv(obj, Unit(17))

	@pytest.mark.parametrize("obj", one2thousand)
	def test_floordiv(self, obj: int):
		assert floordiv(Unit(17), obj) == 17 // obj
		assert floordiv(Unit(17 / 2), obj) == (17 / 2) // obj

	@pytest.mark.parametrize("obj", units_zero2thousand)
	def test_floordiv_errors(self, obj: Unit):
		with pytest.raises(NotImplementedError, match="Dividing a unit by another unit is not allowed."):
			floordiv(Unit(17), obj)
		with pytest.raises(NotImplementedError, match="Dividing a unit by another unit is not allowed."):
			floordiv(obj, Unit(17))

	@pytest.mark.parametrize("obj", units_zero2thousand)
	def test_modulo_errors(self, obj: Unit):
		with pytest.raises(NotImplementedError, match="Modulo division of a unit by another unit is not allowed."):
			Unit(17) % obj  # pylint: disable=expression-not-assigned
		with pytest.raises(NotImplementedError, match="Modulo division of a unit by another unit is not allowed."):
			obj % Unit(17)  # pylint: disable=expression-not-assigned

	@pytest.mark.parametrize("obj", one2thousand)
	def test_modulo(self, obj: int):
		assert Unit(17) % obj == 17 % obj

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_rtruediv_errors(self, obj: int):
		with pytest.raises(NotImplementedError, match="Dividing by a unit is not allowed."):
			truediv(obj, Unit(17))


# TODO eq


@pytest.mark.parametrize("unit", units_of_12)
class TestAdd:

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_add_unit(self, unit: Unit, obj: int):
		assert isinstance(unit + Unit(obj), Unit)
		assert (unit + Unit(obj)).name == unit.name
		assert (unit + Unit(obj)).as_pt() == unit.as_pt() + obj

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_radd_unit(self, unit: Unit, obj: int):
		assert isinstance(Unit(obj) + unit, Unit)
		assert (Unit(obj) + unit).name == "pt"
		assert (Unit(obj) + unit).as_pt() == unit.as_pt() + obj

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_add_number(self, unit: Unit, obj: int):

		assert isinstance(Unit(12) + obj, Unit)
		assert (Unit(12) + obj).name == "pt"
		assert (Unit(12) + obj).as_pt() == 12 + obj

		assert isinstance(UnitInch(1) + obj, Unit)
		assert (UnitInch(1) + obj).name == "inch"
		assert (UnitInch(1) + obj) == UnitInch.from_pt(72) + obj

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_radd_number(self, unit: Unit, obj: int):
		assert isinstance(obj + Unit(12), Unit)
		assert (obj + Unit(12)).name == "pt"
		assert (obj + Unit(12)).as_pt() == 12 + obj

		assert isinstance(obj + UnitInch(1), Unit)
		assert (obj + UnitInch(1)).name == "inch"
		assert (obj + UnitInch(1)) == UnitInch.from_pt(72) + obj


@pytest.mark.parametrize("unit", units_of_12)
class TestSub:

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_sub_unit(self, unit: Unit, obj: int):
		assert isinstance(unit + Unit(obj), Unit)
		assert (unit - Unit(obj)).name == unit.name
		assert (unit - Unit(obj)).as_pt() == unit.as_pt() - obj

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_rsub_unit(self, unit: Unit, obj: int):
		assert isinstance(Unit(obj) - unit, Unit)
		assert (Unit(obj) - unit).name == "pt"
		assert (Unit(obj) - unit).as_pt() == obj - unit.as_pt()

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_sub_number(self, unit: Unit, obj: int):

		assert isinstance(Unit(12) - obj, Unit)
		assert (Unit(12) - obj).name == "pt"
		assert (Unit(12) - obj).as_pt() == 12 - obj

		assert isinstance(UnitInch(1) - obj, Unit)
		assert (UnitInch(1) - obj).name == "inch"
		assert (UnitInch(1) - obj) == UnitInch.from_pt(72) - obj

	@pytest.mark.parametrize("obj", zero2thousand)
	def test_rsub_number(self, unit: Unit, obj: int):
		assert isinstance(obj + Unit(12), Unit)
		assert (obj - Unit(12)).name == "pt"
		assert (obj - Unit(12)).as_pt() == obj - 12

		assert isinstance(obj - UnitInch(1), Unit)
		assert (obj - UnitInch(1)).name == "inch"
		assert (obj - UnitInch(1)) == obj - UnitInch.from_pt(72)