feat: initial otko import

tests/gui/test_assign_load_pattern_name.py

@@ -0,0 +1,121 @@
+"""AssignLoadDialog lets the user pick or name a load pattern."""
+
+from __future__ import annotations
+
+import pytest
+
+pytest.importorskip("PySide6")
+
+from otko.commands import AddNodalLoadsCommand  # noqa: E402
+from otko.core import (  # noqa: E402
+    LinearTimeSeries,
+    Node,
+    PlainLoadPattern,
+    Project,
+)
+from otko.viewmodels import ProjectViewModel  # noqa: E402
+from otko.views.dialogs.assign_load import AssignLoadDialog  # noqa: E402
+
+
+@pytest.mark.gui
+def test_dialog_defaults_to_new_when_no_patterns(qtbot) -> None:  # type: ignore[no-untyped-def]
+    dlg = AssignLoadDialog(n_selected=1, existing_patterns=[])
+    qtbot.addWidget(dlg)
+    # Only one row, the "<New pattern…>" sentinel.
+    assert dlg._pattern_cb.count() == 1
+    assert dlg.selected_pattern_id() is None
+    assert dlg._new_name_edit.isEnabled()
+
+
+@pytest.mark.gui
+def test_dialog_lists_existing_patterns(qtbot) -> None:  # type: ignore[no-untyped-def]
+    dlg = AssignLoadDialog(
+        n_selected=2,
+        existing_patterns=[(1, "Gravity"), (2, "RefMoment")],
+    )
+    qtbot.addWidget(dlg)
+    # 2 existing + 1 new sentinel.
+    assert dlg._pattern_cb.count() == 3
+    assert dlg._pattern_cb.currentIndex() == 0
+    assert dlg.selected_pattern_id() == 1
+    # Switch to RefMoment.
+    dlg._pattern_cb.setCurrentIndex(1)
+    assert dlg.selected_pattern_id() == 2
+    # The new-name field is disabled for existing picks.
+    assert not dlg._new_name_edit.isEnabled()
+
+
+@pytest.mark.gui
+def test_dialog_enables_name_field_for_new_pattern(qtbot) -> None:  # type: ignore[no-untyped-def]
+    dlg = AssignLoadDialog(
+        n_selected=1, existing_patterns=[(1, "Gravity")],
+    )
+    qtbot.addWidget(dlg)
+    dlg._pattern_cb.setCurrentIndex(1)   # "<New pattern…>"
+    assert dlg.selected_pattern_id() is None
+    assert dlg._new_name_edit.isEnabled()
+    dlg._new_name_edit.setText("RefMoment")
+    assert dlg.new_pattern_name() == "RefMoment"
+
+
+@pytest.mark.gui
+def test_command_creates_named_pattern(qtbot) -> None:  # type: ignore[no-untyped-def]
+    """AddNodalLoadsCommand with new_pattern_name must create a fresh
+    PlainLoadPattern with that name, rather than reusing a default."""
+    vm = ProjectViewModel()
+    vm.new_project(ndm=2, ndf=3)
+    vm.project.nodes.append(Node(id=1, coords=(0, 0, 0)))
+    # Pre-existing 'Gravity' pattern — must NOT be reused when the user
+    # asks for a new one named 'RefMoment'.
+    vm.project.time_series.append(LinearTimeSeries(id=1, name="Gravity"))
+    vm.project.load_patterns.append(PlainLoadPattern(
+        id=1, name="Gravity", time_series_id=1,
+    ))
+
+    vm.apply_command(AddNodalLoadsCommand(
+        vm, {1}, (0, 0, 0, 0, 0, 1.0),
+        pattern_id=None, new_pattern_name="RefMoment",
+    ))
+    # Two patterns now: Gravity (id=1) and RefMoment (id=2).
+    names = [p.name for p in vm.project.load_patterns]
+    assert "Gravity" in names and "RefMoment" in names
+    ref = next(p for p in vm.project.load_patterns if p.name == "RefMoment")
+    assert ref.id == 2
+    assert ref.nodal_loads[0].forces[5] == pytest.approx(1.0)
+
+
+@pytest.mark.gui
+def test_command_creates_constant_timeseries_when_requested(qtbot) -> None:  # type: ignore[no-untyped-def]
+    """new_ts_type='Constant' produces a ConstantTimeSeries alongside
+    the new pattern — required for axial preloads in moment-curvature
+    runs where pseudoTime must NOT scale the axial force."""
+    from otko.core import ConstantTimeSeries
+    vm = ProjectViewModel()
+    vm.new_project(ndm=2, ndf=3)
+    vm.project.nodes.append(Node(id=1, coords=(0, 0, 0)))
+    vm.apply_command(AddNodalLoadsCommand(
+        vm, {1}, (-180, 0, 0, 0, 0, 0),
+        pattern_id=None, new_pattern_name="AxialP", new_ts_type="Constant",
+    ))
+    # Pattern + TimeSeries both created with the "AxialP" name.
+    assert vm.project.load_patterns[0].name == "AxialP"
+    ts = vm.project.time_series[0]
+    assert isinstance(ts, ConstantTimeSeries)
+    assert ts.name == "AxialP"
+
+
+@pytest.mark.gui
+def test_command_reuses_existing_pattern_by_id(qtbot) -> None:  # type: ignore[no-untyped-def]
+    """pattern_id=1 reuses the existing pattern — doesn't create new."""
+    vm = ProjectViewModel()
+    vm.new_project(ndm=2, ndf=3)
+    vm.project.nodes.append(Node(id=1, coords=(0, 0, 0)))
+    vm.project.time_series.append(LinearTimeSeries(id=1, name="Existing"))
+    vm.project.load_patterns.append(PlainLoadPattern(
+        id=1, name="Existing", time_series_id=1,
+    ))
+    vm.apply_command(AddNodalLoadsCommand(
+        vm, {1}, (0, 0, 0, 0, 0, 5.0), pattern_id=1,
+    ))
+    assert len(vm.project.load_patterns) == 1
+    assert len(vm.project.load_patterns[0].nodal_loads) == 1