smillmorel
61b850bf46
Some checks failed
CI / lint (pull_request) Has been cancelled
CI / type (pull_request) Has been cancelled
CI / test-headless (macos-latest, 3.10) (pull_request) Has been cancelled
CI / test-headless (macos-latest, 3.11) (pull_request) Has been cancelled
CI / test-headless (macos-latest, 3.12) (pull_request) Has been cancelled
CI / test-headless (ubuntu-latest, 3.10) (pull_request) Has been cancelled
CI / test-headless (ubuntu-latest, 3.11) (pull_request) Has been cancelled
CI / test-headless (ubuntu-latest, 3.12) (pull_request) Has been cancelled
CI / test-headless (windows-latest, 3.10) (pull_request) Has been cancelled
CI / test-headless (windows-latest, 3.11) (pull_request) Has been cancelled
CI / test-headless (windows-latest, 3.12) (pull_request) Has been cancelled
CI / test-gui (pull_request) Has been cancelled
CI / test-integration (macos-latest) (pull_request) Has been cancelled
CI / test-integration (ubuntu-latest) (pull_request) Has been cancelled
CI / test-integration (windows-latest) (pull_request) Has been cancelled
Auto-fixes (UP037, UP038, F401, I001, SIM105, RUF100) plus hand-fixes: drop the now-unused `Union` imports, annotate mutable class attributes with ClassVar, raise ValidationError instead of blind Exception in the zero-length test, delete two dead locals, and bind the per-iteration grid-builder closures to their loop variables (B023).
120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
"""Tests for DrawTrussTool — two-click truss-bar drawing."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
pytest.importorskip("PySide6")
|
|
|
|
from otko.core import (
|
|
CoordinateGridSystem,
|
|
ElasticUniaxial,
|
|
GridSystem,
|
|
Node,
|
|
TrussElement,
|
|
make_grid_lines,
|
|
)
|
|
from otko.viewmodels import ProjectViewModel
|
|
from otko.views.canvas3d.selection import SelectionState
|
|
from otko.views.tools.draw_truss import DrawTrussTool
|
|
|
|
|
|
class _CanvasStub:
|
|
def __init__(self) -> None:
|
|
self.selection = SelectionState()
|
|
|
|
def view_xy(self) -> None:
|
|
pass
|
|
|
|
def set_snap_preview_enabled(self, _enabled: bool) -> None:
|
|
pass
|
|
|
|
|
|
def _vm_with_grid_and_nodes() -> ProjectViewModel:
|
|
vm = ProjectViewModel()
|
|
vm.new_project()
|
|
vm.project.nodes.extend(
|
|
[
|
|
Node(id=1, coords=(0, 0, 0)),
|
|
Node(id=2, coords=(3, 0, 0)),
|
|
]
|
|
)
|
|
vm.project.coord_systems = [
|
|
CoordinateGridSystem(
|
|
name="Global",
|
|
grid=GridSystem(
|
|
x_grid_lines=make_grid_lines("X", [0.0, 3.0]),
|
|
y_grid_lines=make_grid_lines("Y", [0.0]),
|
|
z_grid_lines=make_grid_lines("Z", [0.0]),
|
|
),
|
|
),
|
|
]
|
|
return vm
|
|
|
|
|
|
@pytest.mark.gui
|
|
def test_two_clicks_create_truss(qtbot) -> None: # type: ignore[no-untyped-def]
|
|
vm = _vm_with_grid_and_nodes()
|
|
tool = DrawTrussTool(_CanvasStub(), vm) # type: ignore[arg-type]
|
|
tool.activate()
|
|
|
|
tool.on_node_picked(1)
|
|
tool.on_node_picked(2)
|
|
|
|
assert len(vm.project.elements) == 1 # type: ignore[union-attr]
|
|
el = vm.project.elements[0] # type: ignore[union-attr]
|
|
assert isinstance(el, TrussElement)
|
|
assert el.nodes == (1, 2)
|
|
assert el.area > 0
|
|
# A default ElasticUniaxial must have been created for the material.
|
|
assert len(vm.project.materials) == 1 # type: ignore[union-attr]
|
|
assert isinstance(vm.project.materials[0], ElasticUniaxial)
|
|
|
|
|
|
@pytest.mark.gui
|
|
def test_draw_truss_reuses_existing_material(qtbot) -> None: # type: ignore[no-untyped-def]
|
|
vm = _vm_with_grid_and_nodes()
|
|
vm.project.materials.append(ElasticUniaxial(id=1, name="Existing", E=150e9)) # type: ignore[union-attr]
|
|
tool = DrawTrussTool(_CanvasStub(), vm) # type: ignore[arg-type]
|
|
tool.activate()
|
|
tool.on_node_picked(1)
|
|
tool.on_node_picked(2)
|
|
el = vm.project.elements[0] # type: ignore[union-attr]
|
|
assert el.material_id == 1
|
|
assert len(vm.project.materials) == 1 # type: ignore[union-attr] (no new material)
|
|
|
|
|
|
@pytest.mark.gui
|
|
def test_empty_click_creates_node_then_truss(qtbot) -> None: # type: ignore[no-untyped-def]
|
|
"""Canvas-assumed: coordinates passed are already on a snap target."""
|
|
vm = ProjectViewModel()
|
|
vm.new_project()
|
|
vm.project.coord_systems = [
|
|
CoordinateGridSystem(
|
|
name="Global",
|
|
grid=GridSystem(
|
|
x_grid_lines=make_grid_lines("X", [0.0, 3.0]),
|
|
y_grid_lines=make_grid_lines("Y", [0.0]),
|
|
z_grid_lines=make_grid_lines("Z", [0.0]),
|
|
),
|
|
),
|
|
]
|
|
tool = DrawTrussTool(_CanvasStub(), vm) # type: ignore[arg-type]
|
|
tool.activate()
|
|
tool.on_empty_clicked(0.0, 0.0, 0.0)
|
|
tool.on_empty_clicked(3.0, 0.0, 0.0)
|
|
|
|
# Two nodes created + one truss element.
|
|
assert len(vm.project.nodes) == 2 # type: ignore[union-attr]
|
|
assert len(vm.project.elements) == 1 # type: ignore[union-attr]
|
|
assert isinstance(vm.project.elements[0], TrussElement)
|
|
|
|
|
|
@pytest.mark.gui
|
|
def test_self_pick_is_ignored(qtbot) -> None: # type: ignore[no-untyped-def]
|
|
vm = _vm_with_grid_and_nodes()
|
|
tool = DrawTrussTool(_CanvasStub(), vm) # type: ignore[arg-type]
|
|
tool.activate()
|
|
tool.on_node_picked(1)
|
|
tool.on_node_picked(1) # same node — must not create a zero-length truss
|
|
assert vm.project.elements == [] # type: ignore[union-attr]
|