smillmorel
ba783718d4
Catch-all for the intermixed residue of the unpushed otko-development work ported into this tree: combinations/console-dock/quick-guide wiring across commands, core, services, views and tests; repo-wide ruff-format normalization; README/CONTRIBUTING updates; and the toolbar default (both toolbars now open in the top area, quick guide text updated). Splitting this further would require hunk-level surgery with low confidence; the preceding commits in this branch isolate the self-contained features.
53 lines
1.8 KiB
Python
53 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]
|