freesurfer-surface/tests/test_vertex.py

import numpy
import pytest

from freesurfer_surface import Vertex


def test_init():
    vertex = Vertex(-4.0, 0.5, 21.42)
    assert isinstance(vertex, numpy.ndarray)
    assert vertex.right == pytest.approx(-4.0)
    assert vertex.anterior == pytest.approx(0.5)
    assert vertex.superior == pytest.approx(21.42)


def test_init_kwargs():
    vertex = Vertex(right=-4.0, superior=21.42, anterior=0.5)
    assert vertex.right == pytest.approx(-4.0)
    assert vertex.anterior == pytest.approx(0.5)
    assert vertex.superior == pytest.approx(21.42)


def test_repr():
    vertex = Vertex(right=-4.0, superior=21.42, anterior=0.5)
    assert repr(vertex) == "Vertex(right=-4.0, anterior=0.5, superior=21.42)"


def test_add():
    assert Vertex(-1.5, 4, 2) + Vertex(2, -4.5, 3) == pytest.approx(
        Vertex(0.5, -0.5, 5)
    )


def test_mult():
    assert Vertex(-1.5, 4, 2) * -3 == pytest.approx(Vertex(4.5, -12, -6))


def test_vars():
    attrs = vars(Vertex(-1.5, 4, 2))
    assert len(attrs) == 3
    assert attrs["right"] == pytest.approx(-1.5)
    assert attrs["anterior"] == pytest.approx(4)
    assert attrs["superior"] == pytest.approx(2)


@pytest.mark.parametrize(
    ("vertex_a", "vertex_b", "expected_distance_mm"),
    [
        (Vertex(0, 0, 0), Vertex(0, 0, 0), 0),
        (Vertex(0, 0, 0), Vertex(1, 0, 0), 1),
        (Vertex(0, 0, 0), Vertex(0, 1, 0), 1),
        (Vertex(0, 0, 0), Vertex(0, 0, 1), 1),
        (Vertex(0, 0, 0), Vertex(1, 1, 0), 2 ** (1 / 2)),
        (Vertex(0, 0, 0), Vertex(1, 1, 1), 3 ** (1 / 2)),
        (Vertex(1, 2, 3), Vertex(2, 3, 4), 3 ** (1 / 2)),
        (Vertex(1, 2, 3), Vertex(5, 8, -1), (16 + 36 + 16) ** (1 / 2)),
        (Vertex(0, 0, 0), [Vertex(0, 0, 1), Vertex(0, 0, 2)], [1, 2]),
        (Vertex(0, 0, 0), (Vertex(0, 0, 1), Vertex(0, 0, 2)), [1, 2]),
        (Vertex(0, 0, 0), numpy.vstack((Vertex(0, 0, 1), Vertex(0, 0, 2))), [1, 2]),
        (
            Vertex(1, 2, 3),
            (Vertex(2, 3, 4), Vertex(3, 4, 5)),
            [3 ** (1 / 2), 12 ** (1 / 2)],
        ),
        (
            Vertex(1, 2, 3),
            (Vertex(2, 3, 4), Vertex(3, 4, 5), Vertex(3, 4, 6)),
            [3 ** (1 / 2), 12 ** (1 / 2), 17 ** (1 / 2)],
        ),
    ],
)
def test_distance(vertex_a, vertex_b, expected_distance_mm):
    assert vertex_a.distance_mm(vertex_b) == pytest.approx(expected_distance_mm)