freesurfer-surface/tests/test_polygonal_chain.py

import pytest

from freesurfer_surface import PolygonalChain, PolygonalChainsNotOverlapingError, _LineSegment


def test_init():
    chain = PolygonalChain((0, 21, 42, 84))
    assert tuple(chain.vertex_indices) == (0, 21, 42, 84)


def test_reassign_vertex_indices():
    chain = PolygonalChain((0, 21, 42, 84))
    chain.vertex_indices = (1, 2, 3, 4)
    assert tuple(chain.vertex_indices) == (1, 2, 3, 4)


def test_eq():
    assert PolygonalChain((0, 1, 2)) == PolygonalChain((0, 1, 2))
    # pylint: disable=unneeded-not
    assert not PolygonalChain((0, 1, 2)) == PolygonalChain((0, 1, 4))
    assert not PolygonalChain((0, 1, 2)) == PolygonalChain((0, 4, 2))
    assert not PolygonalChain((0, 1, 2)) == PolygonalChain((4, 1, 2))
    assert not PolygonalChain((0, 1, 2)) == PolygonalChain((1, 2, 3, 4))


def test_repr():
    assert repr(PolygonalChain([])) \
        == 'PolygonalChain(vertex_indices=())'
    assert repr(PolygonalChain((0, 2, 1))) \
        == 'PolygonalChain(vertex_indices=(0, 2, 1))'
    assert repr(PolygonalChain((0, 2, 1, 4, 3))) \
        == 'PolygonalChain(vertex_indices=(0, 2, 1, 4, 3))'


@pytest.mark.parametrize(('vertex_indices_a', 'vertex_indices_b', 'expected_vertex_indices'), [
    ((1, 2, 3), (3, 4), (1, 2, 3, 4)),
    ((1, 2, 3), (4, 3), (1, 2, 3, 4)),
    ((3, 2, 1), (3, 4), (4, 3, 2, 1)),
    ((3, 2, 1), (4, 3), (4, 3, 2, 1)),
    ((1,), (1,), (1,)),
    ((1, 2), (1,), (1, 2)),
    ((1, 2), (2,), (1, 2)),
    ((0, 3, 1, 5, 2), (3, 5, 2, 0), (3, 5, 2, 0, 3, 1, 5, 2)),
    ((98792, 98807, 98821), (98792, 98793), (98793, 98792, 98807, 98821)),
])
def test_connect(vertex_indices_a, vertex_indices_b, expected_vertex_indices):
    chain = PolygonalChain(vertex_indices_a)
    chain.connect(PolygonalChain(vertex_indices_b))
    assert PolygonalChain(expected_vertex_indices) == chain


@pytest.mark.parametrize(('vertex_indices_a', 'vertex_indices_b'), [
    ((1, 2, 3), (2, 4)),
])
def test_connect_fail(vertex_indices_a, vertex_indices_b):
    chain = PolygonalChain(vertex_indices_a)
    with pytest.raises(PolygonalChainsNotOverlapingError):
        chain.connect(PolygonalChain(vertex_indices_b))


@pytest.mark.parametrize(('vertex_indices_a', 'vertex_indices_b'), [
    ((1, 2, 3), ()),
    ((), (3, 4)),
])
def test_connect_fail_empty(vertex_indices_a, vertex_indices_b):
    chain = PolygonalChain(vertex_indices_a)
    with pytest.raises(Exception):
        chain.connect(PolygonalChain(vertex_indices_b))


def test_adjacent_vertex_indices_1():
    chain = PolygonalChain((0, 1, 4, 8))
    pairs = list(chain.adjacent_vertex_indices(1))
    assert len(pairs) == 4
    assert pairs[0] == (0,)
    assert pairs[1] == (1,)
    assert pairs[2] == (4,)
    assert pairs[3] == (8,)


def test_adjacent_vertex_indices_2():
    chain = PolygonalChain((0, 1, 4, 8))
    pairs = list(chain.adjacent_vertex_indices(2))
    assert len(pairs) == 3
    assert pairs[0] == (0, 1)
    assert pairs[1] == (1, 4)
    assert pairs[2] == (4, 8)


def test_adjacent_vertex_indices_3():
    chain = PolygonalChain((0, 1, 4, 8))
    pairs = list(chain.adjacent_vertex_indices(3))
    assert len(pairs) == 2
    assert pairs[0] == (0, 1, 4)
    assert pairs[1] == (1, 4, 8)


def test_segments():
    chain = PolygonalChain((0, 1, 4, 8))
    segments = list(chain.segments())
    assert len(segments) == 3
    assert segments[0] == _LineSegment((0, 1))
    assert segments[1] == _LineSegment((1, 4))
    assert segments[2] == _LineSegment((4, 8))