70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from app.simulation.components.amesim.mechanical.translational import (
|
|
AmesimF000,
|
|
AmesimForc,
|
|
AmesimMecmas21,
|
|
)
|
|
from app.simulation.core.medium import IdealGasMedium
|
|
from app.simulation.registry import COMPONENT_MODEL_REGISTRY
|
|
|
|
|
|
class AmesimMechanicalPublicComponentTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.medium = IdealGasMedium()
|
|
|
|
def test_f000_constrains_mechanical_port_force_to_zero(self) -> None:
|
|
source = AmesimF000("zero_1")
|
|
source.port_1.f = 12.5
|
|
|
|
residuals = source.pressure_flow_equation_residuals()
|
|
|
|
self.assertEqual(len(residuals), 1)
|
|
self.assertEqual(residuals[0].variables, ("zero_1.port_1.f",))
|
|
self.assertAlmostEqual(residuals[0].value, 12.5)
|
|
|
|
def test_forc_converts_signal_to_opposing_source_port_force(self) -> None:
|
|
converter = AmesimForc("force_1")
|
|
converter.res.signal = 20.0
|
|
converter.port_2.f = -20.0
|
|
|
|
residuals = converter.pressure_flow_equation_residuals()
|
|
|
|
self.assertAlmostEqual(converter.output_force, 20.0)
|
|
self.assertAlmostEqual(residuals[0].value, 0.0)
|
|
self.assertEqual(converter.component_result_values(), {"force": 20.0})
|
|
|
|
def test_mecmas21_acceleration_uses_connected_port_forces(self) -> None:
|
|
mass = AmesimMecmas21(
|
|
"mass_1",
|
|
self.medium,
|
|
mass=2.0,
|
|
fcoul=0.0,
|
|
rvisc=0.0,
|
|
wind=0.0,
|
|
x0=0.1,
|
|
v0=0.2,
|
|
)
|
|
mass.port_1.f = 10.0
|
|
mass.port_2.f = -2.0
|
|
|
|
self.assertEqual(mass.get_state_vector(), [0.2, 0.1])
|
|
self.assertAlmostEqual(mass.acceleration(), 4.0)
|
|
self.assertEqual(mass.state_derivative_from_ports({}), [4.0, 0.2])
|
|
self.assertAlmostEqual(mass.port_1.x, 0.1)
|
|
self.assertAlmostEqual(mass.port_2.v, 0.2)
|
|
|
|
def test_mecmas21_registry_rejects_fractional_integer_options(self) -> None:
|
|
with self.assertRaisesRegex(ValueError, "useFriction must be an integer"):
|
|
COMPONENT_MODEL_REGISTRY["amesim_mecmas21"].create(
|
|
"mass_1",
|
|
self.medium,
|
|
{"useFriction": 0.5},
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|