import lsystc as ls
import pytest
from numpy.testing import assert_allclose


@pytest.mark.parametrize("axiom, rules, expected_result", [
    ('A', [('A', 'AB'), ('B', 'BB')], 'AB'),
    ('AB', [('A', 'AB'), ('B', 'BB')], 'ABBB'),
])
def test_single_iteration(axiom, rules, expected_result):
    syst = ls.Lsystc(ls.Config(), axiom, rules, nbiter=1)
    assert syst.dev == expected_result


@pytest.mark.parametrize("axiom, rules, nbiterations, expected_result", [
    ('A', [('A', 'AB'), ('B', 'BB')], 2, 'ABBB'),
    ('AB', [('A', 'AB'), ('B', 'BB')], 3, 'ABBBBBBBBBBBBBBB'),
])
def test_more_iterations(axiom, rules, nbiterations, expected_result):
    syst = ls.Lsystc(ls.Config(), axiom, rules, nbiter=nbiterations)
    assert syst.dev == expected_result


@pytest.mark.parametrize("axiom, expected_result", [
    ('F', [([0, 10], [0, 0], [0, 0], (228, 26, 28))]),
    ('F+F', [([0, 10, 10], [0, 0, 10], [0, 0, 0], (228, 26, 28))]),
])
def test_turt(axiom, expected_result):
    syst = ls.Lsystc(ls.Config(), axiom, [], nbiter=1)
    syst.turtle()
    assert syst.turt == expected_result


@pytest.mark.parametrize("axiom, expected_result", [
    ('⇧F', [([0, 0, 10], [0, 0, 0], [0, 10, 10], (228, 26, 28))]),
    ('⇩F', [([0, 0, 10], [0, 0, 0], [0, -10, -10], (228, 26, 28))]),
])
def test_turt_3d_up_down(axiom, expected_result):
    syst = ls.Lsystc(ls.Config(), axiom, [], nbiter=1)
    syst.turtle()
    assert syst.turt == expected_result


@pytest.mark.parametrize("axiom, expected_result", [
    ('m+F', [([0.0, 0.0], [0.0, 0.0], [0.0, -10.0], (228, 26, 28))]),
    ('p+F', [([0.0, 0.0], [0.0, 0.0], [0.0, 10.0], (228, 26, 28))]),
])
def test_turt_3d_rotation_1(axiom, expected_result):
    syst = ls.Lsystc(ls.Config(), axiom, [], nbiter=1)
    syst.turtle()

    lxr = syst.turt[0][0]
    lxe = expected_result[0][0]

    assert_allclose(lxr, lxe, atol=1e-6)

    lyr = syst.turt[0][1]
    lye = expected_result[0][1]

    assert_allclose(lyr, lye, atol=1e-6)

    lzr = syst.turt[0][2]
    lze = expected_result[0][2]

    assert_allclose(lzr, lze, atol=1e-6)

    lcr = syst.turt[0][3]
    lce = expected_result[0][3]

    assert lcr == lce


@pytest.mark.parametrize("axiom, expected_result", [
    ('MF', [([0.0, 0.0], [0.0, 0.0], [0.0, 10.0], (228, 26, 28))]),
    ('PF', [([0.0, 0.0], [0.0, 0.0], [0.0, -10.0], (228, 26, 28))]),
])
def test_turt_3d_rotation_2(axiom, expected_result):
    syst = ls.Lsystc(ls.Config(), axiom, [], nbiter=1)
    syst.turtle()

    lxr = syst.turt[0][0]
    lxe = expected_result[0][0]

    assert_allclose(lxr, lxe, atol=1e-6)

    lyr = syst.turt[0][1]
    lye = expected_result[0][1]

    assert_allclose(lyr, lye, atol=1e-6)

    lzr = syst.turt[0][2]
    lze = expected_result[0][2]

    assert_allclose(lzr, lze, atol=1e-6)

    lcr = syst.turt[0][3]
    lce = expected_result[0][3]

    assert lcr == lce


@pytest.mark.parametrize("axiom, expected_result", [
    ('uF', [([0, 10.1], [0, 0], [0, 0], (228, 26, 28))]),
    ('vF', [([0, 9.9], [0, 0], [0, 0], (228, 26, 28))]),
])
def test_turt_delta(axiom, expected_result):
    syst = ls.Lsystc(ls.Config(), axiom, [], nbiter=1)
    syst.turtle()
    assert syst.turt == expected_result