smillmorel
612936a00b
Some checks failed
CI / lint (push) Has been cancelled
CI / test (macos-latest, 3.10) (push) Has been cancelled
CI / test (macos-latest, 3.11) (push) Has been cancelled
CI / test (macos-latest, 3.12) (push) Has been cancelled
CI / test (ubuntu-latest, 3.10) (push) Has been cancelled
CI / test (ubuntu-latest, 3.11) (push) Has been cancelled
CI / test (ubuntu-latest, 3.12) (push) Has been cancelled
CI / test (windows-latest, 3.10) (push) Has been cancelled
CI / test (windows-latest, 3.11) (push) Has been cancelled
CI / test (windows-latest, 3.12) (push) Has been cancelled
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
"""Unit tests for the Node entity."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from otko.core import Node
|
|
|
|
|
|
class TestNodeConstruction:
|
|
def test_minimum_required_fields(self) -> None:
|
|
n = Node(id=1, coords=(0.0, 0.0, 0.0))
|
|
assert n.id == 1
|
|
assert n.coords == (0.0, 0.0, 0.0)
|
|
assert n.mass == (0.0,) * 6
|
|
assert n.restraint == (False,) * 6
|
|
assert n.is_free
|
|
assert not n.is_restrained
|
|
|
|
def test_with_name_and_mass(self) -> None:
|
|
n = Node(id=42, name="A1", coords=(1.0, 2.0, 3.0),
|
|
mass=(100.0, 100.0, 100.0, 0.0, 0.0, 0.0))
|
|
assert n.name == "A1"
|
|
assert n.mass[0] == 100.0
|
|
|
|
def test_pin_restraint(self) -> None:
|
|
pin = Node(id=1, coords=(0, 0, 0), restraint=(True, True, True, False, False, False))
|
|
assert pin.is_restrained
|
|
assert sum(pin.restraint) == 3
|
|
|
|
|
|
class TestNodeValidation:
|
|
def test_id_must_be_positive(self) -> None:
|
|
with pytest.raises(ValidationError):
|
|
Node(id=0, coords=(0, 0, 0))
|
|
with pytest.raises(ValidationError):
|
|
Node(id=-1, coords=(0, 0, 0))
|
|
|
|
def test_coords_must_be_three_floats(self) -> None:
|
|
with pytest.raises(ValidationError):
|
|
Node(id=1, coords=(0.0, 0.0)) # type: ignore[arg-type]
|
|
with pytest.raises(ValidationError):
|
|
Node(id=1, coords=(0.0, 0.0, 0.0, 0.0)) # type: ignore[arg-type]
|
|
|
|
def test_mass_must_be_six_components(self) -> None:
|
|
with pytest.raises(ValidationError):
|
|
Node(id=1, coords=(0, 0, 0), mass=(1.0, 2.0, 3.0)) # type: ignore[arg-type]
|
|
|
|
def test_extra_fields_rejected(self) -> None:
|
|
with pytest.raises(ValidationError):
|
|
Node(id=1, coords=(0, 0, 0), unknown="oops") # type: ignore[call-arg]
|