File size: 3,199 Bytes
5f59ba1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 |
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
|