h2lib-tests 13.1.506__py3-none-any.whl → 13.1.901__py3-none-any.whl

h2lib_tests/conftest.py

@@ -0,0 +1,365 @@
+import pytest
+import numpy as np
+from wetb.hawc2.htc_file import HTCFile
+from h2lib_tests.test_files import tfp
+from h2lib._h2lib import H2Lib
+
+
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_tower():
+    # Start from DTU_10MW_RWT and delete everything except the tower.
+    htc = HTCFile(tfp + "DTU_10_MW/htc/DTU_10MW_RWT.htc")
+    htc.set_name("DTU_10MW_RWT_only_tower")
+    for key1 in htc["new_htc_structure"].keys():
+        if key1.startswith("main_body"):
+            if "tower" not in htc["new_htc_structure"][key1]["name"].values:
+                htc["new_htc_structure"][key1].delete()
+        if key1 == "orientation":
+            for key2 in htc["new_htc_structure"]["orientation"].keys():
+                if key2.startswith("relative"):
+                    htc["new_htc_structure"]["orientation"][key2].delete()
+        if key1 == "constraint":
+            for key2 in htc["new_htc_structure"]["constraint"].keys():
+                if key2 != "fix0":
+                    htc["new_htc_structure"]["constraint"][key2].delete()
+    htc["wind"].delete()
+    htc["aerodrag"].delete()
+    htc["aero"].delete()
+    htc["dll"].delete()
+    htc["output"].delete()
+    # Reduce simulation time.
+    htc.simulation.time_stop = 10.0
+    # Change number of bodies in the tower.
+    htc.new_htc_structure.main_body.nbodies = 3
+    # Save the new file.
+    htc.save()
+    return htc
+
+
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_tower_rotated(write_dtu10mw_only_tower):
+    # Start from the DTU_10MW_RWT_only_tower and rotate the tower.
+    htc = write_dtu10mw_only_tower.copy()
+    htc.set_name("DTU_10MW_RWT_only_tower_rotated")
+    alpha = 30.0
+    htc.new_htc_structure.orientation.base.body_eulerang = [
+        alpha,
+        0.0,
+        0.0,
+    ]
+    htc["wind"].delete()
+    htc["output"].delete()
+    htc.save()
+    return (htc, alpha)
+
+
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_tower_encrypted(write_dtu10mw_only_tower):
+    # Start from the DTU_10MW_RWT_only_tower and then encrypt the tower.
+    htc = write_dtu10mw_only_tower.copy()
+    htc.set_name("DTU_10MW_RWT_only_tower_encrypted")
+    # Only the tower is left.
+    htc.new_htc_structure.main_body.timoschenko_input.filename = "./data/DTU_10MW_RWT_Tower_st.dat.v3.enc"
+    htc.save()
+
+
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_blade():
+    # Start from DTU_10MW_RWT and delete everything except the blade.
+    htc = HTCFile(tfp + "DTU_10_MW/htc/DTU_10MW_RWT.htc")
+    htc.set_name("DTU_10MW_RWT_only_blade")
+    for key1 in htc["new_htc_structure"].keys():
+        if key1.startswith("main_body"):
+            if "blade1" not in htc["new_htc_structure"][key1]["name"].values:
+                htc["new_htc_structure"][key1].delete()
+        if key1 == "orientation":
+            htc["new_htc_structure"][key1].delete()
+        if key1 == "constraint":
+            htc["new_htc_structure"][key1].delete()
+    htc["wind"].delete()
+    htc["aerodrag"].delete()
+    htc["aero"].delete()
+    htc["dll"].delete()
+    htc["output"].delete()
+
+    # Set the blade horizontal, to maximize gravity loading.
+    htc.new_htc_structure.add_section("orientation")
+    htc.new_htc_structure.orientation.add_section("base")
+    htc.new_htc_structure.orientation.base.mbdy = "blade1"
+    htc.new_htc_structure.orientation.base.inipos = [0.0, 0.0, 0.0]
+    htc.new_htc_structure.orientation.base["mbdy_eulerang"] = [90.0, 0.0, 0.0]
+    htc.new_htc_structure.orientation.base.mbdy_eulerang.comments = "Blade span is horizontal."
+
+    # Clamp the blade.
+    htc.new_htc_structure.add_section("constraint")
+    htc.new_htc_structure.constraint.add_section("fix0")
+    htc.new_htc_structure.constraint.fix0.mbdy = "blade1"
+
+    # Set as many bodies as elements.
+    htc.new_htc_structure.main_body__7.nbodies = 26
+
+    htc.simulation.log_deltat.delete()
+
+    # Do not use static solver, since it will be done during the test.
+    htc.simulation.solvertype = 2
+    htc.simulation.solvertype.comments = ""
+    htc.simulation.initial_condition = 1
+
+    # No output, as we will use add_sensor().
+
+    # Save the new file.
+    htc.save()
+
+    return htc
+
+
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_blade_low_max_iter(write_dtu10mw_only_blade):
+    # Start from the write_dtu10mw_only_blade and thenreduce the number of max iterations,
+    # so that the static solver will not have time to converge.
+    htc = write_dtu10mw_only_blade.copy()
+    htc.set_name("DTU_10MW_RWT_only_blade_low_max_iter")
+    htc.simulation.max_iterations = 1
+    htc.save()
+
+    return htc
+
+
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_blade_rotate_base(write_dtu10mw_only_blade):
+    # Start from the write_dtu10mw_only_blade and then make it rotate by using the base command.
+    # HAWC2 will use the initial condition, but then the blade will not rotate because of the fix0 constraint.
+    # So, running the simulation will show a clamped blade that vibrates.
+    # Rotate at about 9.5 rpm.
+    htc = write_dtu10mw_only_blade.copy()
+    htc.set_name("DTU_10MW_RWT_only_blade_rotate_base")
+    speed = 1.0  # [rad/s]
+    htc.new_htc_structure.orientation.base["mbdy_ini_rotvec_d1"] = [0.0, 1.0, 0.0, speed]
+    htc.new_htc_structure.orientation.base.mbdy_ini_rotvec_d1.comments = f"= {speed * 30.0 / np.pi:.2f} rpm"
+    htc.new_htc_structure.main_body__7.gravity = 0.0
+    htc["wind"].delete()
+    htc["output"].delete()
+    htc.save()
+
+
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_blade_rotate_relative():
+    # Start from DTU_10MW_RWT and delete everything except the blade and hub.
+    # The blade now rotates because of the relative rotation.
+    # Because of the fix1 constraint, the blade will not rotate after time 0.
+    htc = HTCFile(tfp + "DTU_10_MW/htc/DTU_10MW_RWT.htc")
+    htc.set_name("DTU_10MW_RWT_only_blade_rotate_relative")
+    for key1 in ("main_body",     # tower
+                 "main_body__2",  # towertop
+                 "main_body__3",  # shaft
+                 "main_body__5",  # hub2
+                 "main_body__6",  # hub3
+                 "main_body__8",  # blade2
+                 "main_body__9",  # blade3
+                 "orientation",
+                 "constraint",
+                 ):
+        htc["new_htc_structure"][key1].delete()
+    htc["wind"].delete()
+    htc["aerodrag"].delete()
+    htc["aero"].delete()
+    htc["dll"].delete()
+    htc["output"].delete()
+
+    # Set the hub as vertical.
+    htc.new_htc_structure.add_section("orientation")
+    htc.new_htc_structure.orientation.add_section("base")
+    htc.new_htc_structure.orientation.base.mbdy = "hub1"
+    htc.new_htc_structure.orientation.base.inipos = [0.0, 0.0, 0.0]
+    htc.new_htc_structure.orientation.base["mbdy_eulerang"] = [180.0, 0.0, 0.0]
+    htc.new_htc_structure.orientation.base.mbdy_eulerang.comments = "Hub axis is up."
+
+    # Set the blade horizontal.
+    htc.new_htc_structure.orientation.add_section("relative")
+    htc.new_htc_structure.orientation.relative.mbdy1 = "hub1  last"
+    htc.new_htc_structure.orientation.relative.mbdy2 = "blade1  1"
+    htc.new_htc_structure.orientation.relative["mbdy2_eulerang"] = [-90.0, 0.0, 0.0]
+    htc.new_htc_structure.orientation.relative.mbdy2_eulerang.comments = "Blade span is horizontal."
+    speed = 1.0  # [rad/s]
+    htc.new_htc_structure.orientation.relative["mbdy2_ini_rotvec_d1"] = [0.0, 1.0, 0.0, speed]
+    htc.new_htc_structure.orientation.relative.mbdy2_ini_rotvec_d1.comments = f"= {speed * 30.0 / np.pi:.2f} rpm"
+
+    # Disable gravity.
+    htc.new_htc_structure.main_body__7.gravity = 0.0
+    htc.new_htc_structure.main_body__4.gravity = 0.0
+
+    # Clamp the hub and blade.
+    htc.new_htc_structure.add_section("constraint")
+    htc.new_htc_structure.constraint.add_section("fix0")
+    htc.new_htc_structure.constraint.fix0.mbdy = "hub1"
+    htc.new_htc_structure.constraint.add_section("fix1")
+    htc.new_htc_structure.constraint.fix1.mbdy1 = "hub1  last"
+    htc.new_htc_structure.constraint.fix1.mbdy2 = "blade1  1"
+
+    # Set as many bodies as elements.
+    htc.new_htc_structure.main_body__7.nbodies = 26
+
+    htc.simulation.log_deltat.delete()
+
+    # Do not use static solver, since it will be done during the test.
+    htc.simulation.solvertype = 2
+    htc.simulation.solvertype.comments = ""
+    htc.simulation.initial_condition = 1
+
+    # Save the new file.
+    htc.save()
+
+    return htc
+
+
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_blade_rotate_bearing3():
+    # Start from DTU_10MW_RWT and delete everything except the blade and hub.
+    # The blade now rotates because of the bearing3 between the blade and hub.
+    htc = HTCFile(tfp + "DTU_10_MW/htc/DTU_10MW_RWT.htc")
+    htc.set_name("DTU_10MW_RWT_only_blade_rotate_bearing3")
+    for key1 in ("main_body",     # tower
+                 "main_body__2",  # towertop
+                 "main_body__3",  # shaft
+                 "main_body__5",  # hub2
+                 "main_body__6",  # hub3
+                 "main_body__8",  # blade2
+                 "main_body__9",  # blade3
+                 "orientation",
+                 "constraint",
+                 ):
+        htc["new_htc_structure"][key1].delete()
+    htc["wind"].delete()
+    htc["aerodrag"].delete()
+    htc["aero"].delete()
+    htc["dll"].delete()
+    htc["output"].delete()
+
+    # Set the hub as vertical.
+    htc.new_htc_structure.add_section("orientation")
+    htc.new_htc_structure.orientation.add_section("base")
+    htc.new_htc_structure.orientation.base.mbdy = "hub1"
+    htc.new_htc_structure.orientation.base.inipos = [0.0, 0.0, 0.0]
+    htc.new_htc_structure.orientation.base["mbdy_eulerang"] = [180.0, 0.0, 0.0]
+    htc.new_htc_structure.orientation.base.mbdy_eulerang.comments = "Hub axis is up."
+
+    # Set the blade horizontal.
+    htc.new_htc_structure.orientation.add_section("relative")
+    htc.new_htc_structure.orientation.relative.mbdy1 = "hub1  last"
+    htc.new_htc_structure.orientation.relative.mbdy2 = "blade1  1"
+    htc.new_htc_structure.orientation.relative["mbdy2_eulerang"] = [-90.0, 0.0, 0.0]
+    htc.new_htc_structure.orientation.relative.mbdy2_eulerang.comments = "Blade span is horizontal."
+
+    # Disable gravity.
+    htc.new_htc_structure.main_body__7.gravity = 0.0
+    htc.new_htc_structure.main_body__4.gravity = 0.0
+
+    # Clamp the hub.
+    htc.new_htc_structure.add_section("constraint")
+    htc.new_htc_structure.constraint.add_section("fix0")
+    htc.new_htc_structure.constraint.fix0.mbdy = "hub1"
+
+    # Insert bearing3.
+    htc.new_htc_structure.constraint.add_section("bearing3")
+    htc.new_htc_structure.constraint.bearing3.name = "bearing"
+    htc.new_htc_structure.constraint.bearing3.mbdy1 = "hub1 last"
+    htc.new_htc_structure.constraint.bearing3.mbdy2 = "blade1 1"
+    htc.new_htc_structure.constraint.bearing3.bearing_vector = [1, 0.0, 0.0, 1.0]
+    speed = 1.0  # [rad/s]
+    htc.new_htc_structure.constraint.bearing3.omegas = speed
+    htc.new_htc_structure.constraint.bearing3.omegas.comments = f"= {speed * 30.0 / np.pi:.2f} rpm"
+
+    # Set as many bodies as elements.
+    htc.new_htc_structure.main_body__7.nbodies = 26
+
+    htc.simulation.log_deltat.delete()
+
+    # Do not use static solver, since it will be done during the test.
+    htc.simulation.solvertype = 2
+    htc.simulation.solvertype.comments = ""
+    htc.simulation.initial_condition = 1
+
+    # Save the new file.
+    htc.save()
+
+    return htc
+
+
+@pytest.fixture(scope="session")
+def h2_dtu_10mw_only_tower(write_dtu10mw_only_tower):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_tower.htc"
+    h2.init(htc_path=htc_path, model_path=model_path)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_dtu_10mw_only_tower_rotated(write_dtu10mw_only_tower_rotated):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_tower_rotated.htc"
+    h2.init(htc_path=htc_path, model_path=model_path)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_dtu_10mw_only_tower_encrypted(write_dtu10mw_only_tower_encrypted):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_tower_encrypted.htc"
+    h2.init(htc_path=htc_path, model_path=model_path)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_dtu_10mw_only_blade(write_dtu10mw_only_blade):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_blade.htc"
+    h2.init(htc_path=htc_path, model_path=model_path)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_dtu10mw_only_blade_low_max_iter(write_dtu10mw_only_blade_low_max_iter):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_blade_low_max_iter.htc"
+    h2.init(htc_path=htc_path, model_path=model_path)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_dtu_10mw_only_blade_rotate_base(write_dtu10mw_only_blade_rotate_base):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_blade_rotate_base.htc"
+    h2.init(htc_path=htc_path, model_path=model_path)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_dtu_10mw_only_blade_rotate_relative(write_dtu10mw_only_blade_rotate_relative):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_blade_rotate_relative.htc"
+    h2.init(htc_path=htc_path, model_path=model_path)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_dtu_10mw_only_blade_rotate_bearing3(write_dtu10mw_only_blade_rotate_bearing3):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_blade_rotate_bearing3.htc"
+    h2.init(htc_path=htc_path, model_path=model_path)
+    yield h2
+    h2.close()

h2lib_tests/test_lin.py

@@ -0,0 +1,191 @@
+# -*- coding: utf-8 -*-
+"""
+Test the system-eigenanalysis.
+
+@author: ricriv
+"""
+
+# Here we are relying on the default behavior of pytest, which is to execute
+# the tests in the same order that they are specified.
+# If one day this will not be the case anymore, we can enforce the order by
+# using the solution proposed at: https://stackoverflow.com/a/77793427/3676517
+
+import pytest
+
+import numpy as np
+from numpy import testing as npt
+
+from h2lib._h2lib import H2Lib
+from h2lib_tests.test_files import tfp
+
+
+def test_system_not_linearized_1(h2_dtu_10mw_only_blade):
+    with pytest.raises(RuntimeError, match="SYSTEM_NOT_LINEARIZED"):
+        h2_dtu_10mw_only_blade.do_system_eigenanalysis(2)
+
+
+def test_system_not_linearized_2(h2_dtu_10mw_only_blade):
+    with pytest.raises(RuntimeError, match="SYSTEM_NOT_LINEARIZED"):
+        h2_dtu_10mw_only_blade.get_system_eigenvalues_and_eigenvectors(2, 2)
+
+
+def test_system_not_linearized_3(h2_dtu_10mw_only_blade):
+    with pytest.raises(RuntimeError, match="SYSTEM_NOT_LINEARIZED"):
+        h2_dtu_10mw_only_blade.get_system_matrices(312, 156)
+
+
+def test_linearize(h2_dtu_10mw_only_blade):
+    n_tdofs, n_rdofs = h2_dtu_10mw_only_blade.linearize()
+    assert n_rdofs == 26 * 6  # = number of nodes * 6
+    assert n_tdofs == 26 * 6 * 2  # times 2 because of speed.
+
+
+def test_too_many_modes(h2_dtu_10mw_only_blade):
+    with pytest.raises(ValueError, match="TOO_MANY_MODES_REQUESTED"):
+        h2_dtu_10mw_only_blade.do_system_eigenanalysis(n_modes=1000)
+
+
+def test_sys_eig_not_done(h2_dtu_10mw_only_blade):
+    with pytest.raises(RuntimeError, match="SYSTEM_EIGENANALYSIS_NOT_DONE"):
+        h2_dtu_10mw_only_blade.get_system_eigenvalues_and_eigenvectors(2, 2)
+
+
+def test_sys_eig_no_damping(h2_dtu_10mw_only_blade):
+    natural_frequencies = h2_dtu_10mw_only_blade.do_system_eigenanalysis(
+        n_modes=4, include_damping=False
+    )
+    # Test against: result at the time of writing.
+    npt.assert_allclose(
+        natural_frequencies,
+        np.array([0.610409, 0.930466, 1.739094, 2.761632]),
+        rtol=1e-6,
+    )
+
+
+def test_sys_eig_no_damping_wrong_n_modes(h2_dtu_10mw_only_blade):
+    with pytest.raises(ValueError, match="TOO_MANY_MODES_REQUESTED"):
+        h2_dtu_10mw_only_blade.get_system_eigenvalues_and_eigenvectors(
+            1000, 156
+        )
+
+
+def test_sys_eig_no_damping_wrong_ny(h2_dtu_10mw_only_blade):
+    with pytest.raises(ValueError, match="WRONG_REDUCED_DOF"):
+        h2_dtu_10mw_only_blade.get_system_eigenvalues_and_eigenvectors(4, 2)
+
+
+def test_sys_eig_no_damping_eigv(h2_dtu_10mw_only_blade):
+    n_modes = 4
+    n_rdofs = 156
+    eig_val, eig_vec = (
+        h2_dtu_10mw_only_blade.get_system_eigenvalues_and_eigenvectors(
+            n_modes, n_rdofs, include_damping=False
+        )
+    )
+    assert eig_val.size == n_modes
+    assert eig_vec.shape == (n_modes, n_rdofs)
+    assert eig_val.dtype == np.float64
+    assert eig_vec.dtype == np.float64
+    npt.assert_allclose(
+        eig_val,
+        np.array([3.83531, 5.846293, 10.927047, 17.351849]),
+    )
+
+
+def test_sys_eig_with_damping(h2_dtu_10mw_only_blade):
+    freq, damp = h2_dtu_10mw_only_blade.do_system_eigenanalysis(
+        n_modes=4, include_damping=True
+    )
+    # Test against: result at the time of writing.
+    npt.assert_allclose(
+        freq, np.array([0.610409, 0.930468, 1.739098, 2.761702]), rtol=1e-6
+    )
+    npt.assert_allclose(
+        damp, np.array([0.004826, 0.004758, 0.013395, 0.014194]), atol=1e-6
+    )
+
+
+def test_sys_eig_with_damping_eigv(h2_dtu_10mw_only_blade):
+    n_modes = 4
+    n_rdofs = 156
+    eig_val, eig_vec = (
+        h2_dtu_10mw_only_blade.get_system_eigenvalues_and_eigenvectors(
+            n_modes, n_rdofs, include_damping=True
+        )
+    )
+    assert eig_val.size == n_modes
+    assert eig_vec.shape == (n_modes, 2 * n_rdofs)
+    assert eig_val.dtype == np.complex128
+    assert eig_vec.dtype == np.complex128
+    npt.assert_allclose(
+        eig_val,
+        np.array(
+            [
+                -0.01851 - 3.835266j,
+                -0.027817 - 5.846237j,
+                -0.146367 - 10.926094j,
+                -0.246296 - 17.350536j,
+            ]
+        ),
+        atol=1e-6,
+    )
+
+
+def test_get_system_matrices_wrong_nt(h2_dtu_10mw_only_blade):
+    with pytest.raises(ValueError, match="WRONG_TOTAL_DOF"):
+        h2_dtu_10mw_only_blade.get_system_matrices(300, 156)
+
+
+def test_get_system_matrices_wrong_nr(h2_dtu_10mw_only_blade):
+    with pytest.raises(ValueError, match="WRONG_REDUCED_DOF"):
+        h2_dtu_10mw_only_blade.get_system_matrices(312, 150)
+
+
+def test_get_system_matrices(h2_dtu_10mw_only_blade):
+    n_rdofs = 26 * 6
+    n_tdofs = 2 * n_rdofs
+    M, C, K, R = h2_dtu_10mw_only_blade.get_system_matrices(n_tdofs, n_rdofs)
+    assert M.shape == (n_rdofs, n_rdofs)
+    assert C.shape == (n_rdofs, n_rdofs)
+    assert K.shape == (n_rdofs, n_rdofs)
+    assert R.shape == (n_tdofs, n_rdofs)
+
+
+def test_sys_eig_encrypted(h2_dtu_10mw_only_tower_encrypted):
+    n_tdofs, n_rdofs = h2_dtu_10mw_only_tower_encrypted.linearize()
+    n_modes = 4
+    freq, damp = h2_dtu_10mw_only_tower_encrypted.do_system_eigenanalysis(
+        n_modes=n_modes, include_damping=True
+    )
+    eig_val = h2_dtu_10mw_only_tower_encrypted.get_system_eigenvalues_and_eigenvectors(
+        n_modes=n_modes, n_rdofs=n_rdofs, include_damping=True
+    )
+    npt.assert_allclose(
+        freq,
+        np.array([0.770592, 0.770592, 3.449993, 3.449993]),
+        rtol=1e-6,
+    )
+
+    npt.assert_allclose(
+        damp,
+        np.array([0.010006, 0.010006, 0.044675, 0.044675]),
+        atol=1e-6,
+    )
+    npt.assert_allclose(
+        eig_val,
+        np.array(
+            [
+                -0.048444 - 4.84153j,
+                -0.048444 - 4.84153j,
+                -0.968409 - 21.6553j,
+                -0.968409 - 21.6553j,
+            ]
+        ),
+        atol=1e-6,
+    )
+
+
+def test_test_get_system_matrices_encrypted(h2_dtu_10mw_only_tower_encrypted):
+    n_tdofs, n_rdofs = h2_dtu_10mw_only_tower_encrypted.linearize()
+    with pytest.raises(RuntimeError, match="STRUCTURE_IS_CONFIDENTIAL"):
+        h2_dtu_10mw_only_tower_encrypted.get_system_matrices(n_tdofs, n_rdofs)

h2lib_tests/test_static_solver.py

@@ -18,31 +18,6 @@ from numpy import testing as npt
 from h2lib._h2lib import H2Lib
 from h2lib_tests.test_files import tfp
 
-from .test_write_htc import (
-    write_dtu10mw_only_blade,
-    write_dtu10mw_only_blade_low_max_iter,
-)
-
-
-@pytest.fixture(scope="module")
-def h2_dtu_10mw_only_blade(write_dtu10mw_only_blade):
-    h2 = H2Lib(suppress_output=True)
-    model_path = f"{tfp}DTU_10_MW/"
-    htc_path = "htc/DTU_10MW_RWT_only_blade.htc"
-    h2.init(htc_path=htc_path, model_path=model_path)
-    yield h2
-    h2.close()
-
-
-@pytest.fixture(scope="module")
-def h2_dtu10mw_only_blade_low_max_iter(write_dtu10mw_only_blade_low_max_iter):
-    h2 = H2Lib(suppress_output=True)
-    model_path = f"{tfp}DTU_10_MW/"
-    htc_path = "htc/DTU_10MW_RWT_only_blade_low_max_iter.htc"
-    h2.init(htc_path=htc_path, model_path=model_path)
-    yield h2
-    h2.close()
-
 
 def test_solver_static_update_no_init(h2_dtu_10mw_only_blade):
     with pytest.raises(RuntimeError, match="STATIC_SOLVER_NOT_INITIALIZED"):
@@ -77,7 +52,13 @@ def test_solver_static_delete(h2_dtu_10mw_only_blade):
     h2_dtu_10mw_only_blade.solver_static_delete()
 
 
-def test_static_solver_run(h2_dtu_10mw_only_blade):
+def test_static_solver_run_fail(h2_dtu10mw_only_blade_low_max_iter):
+    with pytest.raises(RuntimeError, match="STATIC_SOLVER_DID_NOT_CONVERGE"):
+        h2_dtu10mw_only_blade_low_max_iter.solver_static_run()
+
+
+def test_static_solver_run_1(h2_dtu_10mw_only_blade):
+    # Use gravity to deflect the clamped blade.
     # Add a sensor for the blade root moment, in this case only due to gravity.
     id = h2_dtu_10mw_only_blade.add_sensor("mbdy momentvec blade1 1 1 blade1")
 
@@ -93,6 +74,52 @@ def test_static_solver_run(h2_dtu_10mw_only_blade):
     )
 
 
-def test_static_solver_run_fail(h2_dtu10mw_only_blade_low_max_iter):
-    with pytest.raises(RuntimeError, match="STATIC_SOLVER_DID_NOT_CONVERGE"):
-        h2_dtu10mw_only_blade_low_max_iter.solver_static_run()
+def test_static_solver_run_2(h2_dtu_10mw_only_blade_rotate_base):
+    # Apply centrifugal loading with the base command.
+    # Add a sensor for the blade root force, in this case only due to centrifugal force.
+    id = h2_dtu_10mw_only_blade_rotate_base.add_sensor("mbdy forcevec blade1 1 1 blade1")
+
+    # Run the static solver.
+    h2_dtu_10mw_only_blade_rotate_base.solver_static_run()
+
+    # Do 1 step to get the output.
+    h2_dtu_10mw_only_blade_rotate_base.step()
+    val = h2_dtu_10mw_only_blade_rotate_base.get_sensor_values(id)
+    # Test against: result at the time of writing.
+    npt.assert_allclose(
+        val, np.array([10879.129998, 383.43168, 989.68145])
+    )
+
+
+def test_static_solver_run_3(h2_dtu_10mw_only_blade_rotate_relative):
+    # Apply centrifugal loading with the relative command.
+    # Add a sensor for the blade root force, in this case only due to centrifugal force.
+    id = h2_dtu_10mw_only_blade_rotate_relative.add_sensor("mbdy forcevec blade1 1 1 blade1")
+
+    # Run the static solver.
+    h2_dtu_10mw_only_blade_rotate_relative.solver_static_run()
+
+    # Do 1 step to get the output.
+    h2_dtu_10mw_only_blade_rotate_relative.step()
+    val = h2_dtu_10mw_only_blade_rotate_relative.get_sensor_values(id)
+    # Test against: result at the time of writing.
+    npt.assert_allclose(
+        val, np.array([10879.083576, 383.430039, 989.67935])
+    )
+
+
+def test_static_solver_run_4(h2_dtu_10mw_only_blade_rotate_bearing3):
+    # Apply centrifugal loading with the bearing3 command.
+    # Add a sensor for the blade root moment, in this case only due to centrifugal force.
+    id = h2_dtu_10mw_only_blade_rotate_bearing3.add_sensor("mbdy momentvec blade1 1 1 blade1")
+
+    # Run the static solver.
+    h2_dtu_10mw_only_blade_rotate_bearing3.solver_static_run()
+
+    # Do 1 step to get the output.
+    h2_dtu_10mw_only_blade_rotate_bearing3.step()
+    val = h2_dtu_10mw_only_blade_rotate_bearing3.get_sensor_values(id)
+    # Test against: result at the time of writing.
+    npt.assert_allclose(
+        val, np.array([-3094.986918, 115414.095937, 325.296806])
+    )

h2lib_tests/test_topology_h2lib.py

@@ -4,41 +4,6 @@ from numpy import testing as npt
 from h2lib_tests.test_files import tfp
 import numpy as np
 import pytest
-from .test_write_htc import (
-    write_dtu10mw_only_tower,
-    write_dtu10mw_only_tower_rotated,
-    write_dtu10mw_only_tower_encrypted
-)
-
-
-@pytest.fixture(scope="module")
-def h2_dtu_10mw_only_tower(write_dtu10mw_only_tower):
-    h2 = H2Lib(suppress_output=True)
-    model_path = f"{tfp}DTU_10_MW/"
-    htc_path = "htc/DTU_10MW_RWT_only_tower.htc"
-    h2.init(htc_path=htc_path, model_path=model_path)
-    yield h2
-    h2.close()
-
-
-@pytest.fixture(scope="module")
-def h2_dtu_10mw_only_tower_rotated(write_dtu10mw_only_tower_rotated):
-    h2 = H2Lib(suppress_output=True)
-    model_path = f"{tfp}DTU_10_MW/"
-    htc_path = "htc/DTU_10MW_RWT_only_tower_rotated.htc"
-    h2.init(htc_path=htc_path, model_path=model_path)
-    yield h2
-    h2.close()
-
-
-@pytest.fixture(scope="module")
-def h2_dtu_10mw_only_tower_encrypted(write_dtu10mw_only_tower_encrypted):
-    h2 = H2Lib(suppress_output=True)
-    model_path = f"{tfp}DTU_10_MW/"
-    htc_path = "htc/DTU_10MW_RWT_only_tower_encrypted.htc"
-    h2.init(htc_path=htc_path, model_path=model_path)
-    yield h2
-    h2.close()
 
 
 def test_number_of_bodies_and_constraints(
@@ -74,7 +39,9 @@ def test_get_timoshenko_location(
     h2_dtu_10mw_only_tower,
 ):
     # Test first element.
-    l, r1, r12, tes = h2_dtu_10mw_only_tower.get_timoshenko_location(ibdy=0, ielem=0)
+    l, r1, r12, tes = h2_dtu_10mw_only_tower.get_timoshenko_location(
+        ibdy=0, ielem=0
+    )
     assert l - 11.5 < 1e-14
     npt.assert_array_equal(r1, np.array([0.0, 0.0, 0]))
     npt.assert_array_almost_equal_nulp(r12, np.array([0.0, 0.0, -11.5]))
@@ -84,10 +51,14 @@ def test_get_timoshenko_location(
     )
 
     # Test last element.
-    l, r1, r12, tes = h2_dtu_10mw_only_tower.get_timoshenko_location(ibdy=2, ielem=3)
+    l, r1, r12, tes = h2_dtu_10mw_only_tower.get_timoshenko_location(
+        ibdy=2, ielem=3
+    )
     assert l - 12.13 < 1e-14
     npt.assert_array_almost_equal_nulp(r1, np.array([0.0, 0.0, -34.5]))
-    npt.assert_array_almost_equal_nulp(r12, np.array([0.0, 0.0, -12.13]), nulp=3)
+    npt.assert_array_almost_equal_nulp(
+        r12, np.array([0.0, 0.0, -12.13]), nulp=3
+    )
     npt.assert_array_equal(
         tes,
         np.array([[-1.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, -1.0]]),
@@ -112,7 +83,9 @@ def test_get_timoshenko_location_encrypted(
     h2_dtu_10mw_only_tower_encrypted,
 ):
     with pytest.raises(RuntimeError, match="STRUCTURE_IS_CONFIDENTIAL"):
-        h2_dtu_10mw_only_tower_encrypted.get_timoshenko_location(ibdy=0, ielem=0)
+        h2_dtu_10mw_only_tower_encrypted.get_timoshenko_location(
+            ibdy=0, ielem=0
+        )
 
 
 def test_get_body_rotation_tensor_1(h2_dtu_10mw_only_tower):
@@ -141,3 +114,76 @@ def test_get_body_rotation_tensor_body_does_not_exist(h2_dtu_10mw_only_tower):
 def test_get_body_rotation_tensor_encrypted(h2_dtu_10mw_only_tower_encrypted):
     with pytest.raises(RuntimeError, match="STRUCTURE_IS_CONFIDENTIAL"):
         h2_dtu_10mw_only_tower_encrypted.get_body_rotation_tensor(ibdy=0)
+
+
+def test_set_orientation_base_not_found(h2_dtu_10mw_only_tower):
+    with pytest.raises(ValueError, match="MAIN_BODY_NOT_FOUND"):
+        h2_dtu_10mw_only_tower.set_orientation_base(main_body_name="blade")
+
+
+def test_set_orientation_base_1(
+    h2_dtu_10mw_only_tower, h2_dtu_10mw_only_tower_rotated
+):
+    # Start from h2_dtu_10mw_only_tower and rotate the base.
+    # See if it matches h2_dtu_10mw_only_tower_rotated.
+    h2_dtu_10mw_only_tower.set_orientation_base(
+        main_body_name="tower", mbdy_eulerang_table=np.array([30.0, 0.0, 0.0])
+    )
+    amat_desired = h2_dtu_10mw_only_tower_rotated.get_body_rotation_tensor(
+        ibdy=0
+    )
+    amat_actual = h2_dtu_10mw_only_tower.get_body_rotation_tensor(ibdy=0)
+    npt.assert_array_almost_equal_nulp(amat_actual, amat_desired)
+    # Reset orientation.
+    h2_dtu_10mw_only_tower.set_orientation_base(main_body_name="tower")
+
+
+def test_set_orientation_base_with_reset_orientation(
+    h2_dtu_10mw_only_tower_rotated,
+):
+    h2_dtu_10mw_only_tower_rotated.set_orientation_base(
+        main_body_name="tower", reset_orientation=True
+    )
+    amat_actual = h2_dtu_10mw_only_tower_rotated.get_body_rotation_tensor(
+        ibdy=0
+    )
+    npt.assert_array_almost_equal_nulp(amat_actual, np.eye(3))
+    # Reset orientation.
+    h2_dtu_10mw_only_tower_rotated.set_orientation_base(
+        main_body_name="tower", mbdy_eulerang_table=np.array([30.0, 0.0, 0.0])
+    )
+
+
+def test_set_orientation_base_without_reset_orientation(
+    h2_dtu_10mw_only_tower_rotated,
+):
+    h2_dtu_10mw_only_tower_rotated.set_orientation_base(
+        main_body_name="tower",
+        mbdy_eulerang_table=np.array([-30.0, 0.0, 0.0]),
+        reset_orientation=False,
+    )
+    amat_actual = h2_dtu_10mw_only_tower_rotated.get_body_rotation_tensor(
+        ibdy=0
+    )
+    npt.assert_array_almost_equal_nulp(amat_actual, np.eye(3))
+    # Reset orientation.
+    h2_dtu_10mw_only_tower_rotated.set_orientation_base(
+        main_body_name="tower", mbdy_eulerang_table=np.array([30.0, 0.0, 0.0])
+    )
+
+
+def test_set_orientation_base_speed(
+    h2_dtu_10mw_only_blade
+):
+    # Set speed.
+    h2_dtu_10mw_only_blade.set_orientation_base(
+        main_body_name="blade1",
+        reset_orientation=False,
+        mbdy_ini_rotvec_d1=np.array([0.0, 1.0, 0.0, 1.0]),
+    )
+    # TODO: check against h2_dtu_10mw_only_blade_rotate_base.
+    # Reset speed.
+    h2_dtu_10mw_only_blade.set_orientation_base(
+        main_body_name="blade1",
+        reset_orientation=False,
+    )

{h2lib_tests-13.1.506.dist-info → h2lib_tests-13.1.901.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: h2lib-tests
-Version: 13.1.506
-Summary: Tests and test_files for test h2lib (13.1.8)
+Version: 13.1.901
+Summary: Tests and test_files for test h2lib (13.1.9+5-g2026dd7)
 Download-URL: 
 Author: Mads M Pedersen
 Author-email: 

{h2lib_tests-13.1.506.dist-info → h2lib_tests-13.1.901.dist-info}/RECORD

@@ -1,14 +1,15 @@
 h2lib_tests/__init__.py,sha256=VjSqfGg8BzdmSjfSFhJh4hZbYZ_cME7xp9EWFKHQphA,61
+h2lib_tests/conftest.py,sha256=bKGq79oRIYFbGGCEUjNHxrTYJlitHws-IwK0KZ5bm2c,14326
 h2lib_tests/dtu10mw.py,sha256=a7SXfyDwDQPastYKb5CgghOQcYfgO1eTwGrd-H3Enok,4374
 h2lib_tests/test_calc.py,sha256=VNLfr2J9R2Jy9xTbdZ9dfbQ4dCwr7H7nbZRI3yP69fQ,2152
 h2lib_tests/test_ellipsys_couplings.py,sha256=69I_2h_PMfuIUgxnlLp_6bkS5IEpWMWnnPV1sRrWEp0,2399
 h2lib_tests/test_h2lib.py,sha256=ZKp8pfQQpCUQ7IBXypTmvewbJwTQowSiCEt6t4R9TgY,13384
 h2lib_tests/test_h2rotor.py,sha256=kaE9AFDYSPaMFSgBAbeegd7-l6Z_4_BsQaWie0k32q4,6210
+h2lib_tests/test_lin.py,sha256=FxuDvPuu1MjWE4B7dIWC3qR-9NtjqodGPO1aurUDBwY,6362
 h2lib_tests/test_mpi.py,sha256=emBgXRAvvFFOsVrAziiQCUZvEF9HS5Wc-x4KqqltQP0,6835
 h2lib_tests/test_multiprocessinterface.py,sha256=h2o4havtK6IuMXsplNjGUa3VxOnbpEYGxdrrAKQilj0,1470
-h2lib_tests/test_static_solver.py,sha256=Ik_CgGiFVibEPq0dGzSfSYH3gv_dTm-qcz6JKmJiH_A,3228
-h2lib_tests/test_topology_h2lib.py,sha256=cClNtAIc1wR_gZa7eDO4ZySvhScqCEIHVMKZEhk29QE,4980
-h2lib_tests/test_write_htc.py,sha256=3KMHffkR8A6C1RXODxq_O522fEBfJwp3YpjDmUmiNjU,4774
+h2lib_tests/test_static_solver.py,sha256=OiabTF7IwIskkvBslp32c5o4v53YHQNTTV3vOU5vVb8,4679
+h2lib_tests/test_topology_h2lib.py,sha256=OJj61anq0xT0X-LDoGTYk0E0kkB-MbkxafQWg31Oiys,6527
 h2lib_tests/test_files/__init__.py,sha256=9e6ZUPb42e0wf2E1rutdcTM8hROcWFRVPXtZriU3ySw,50
 h2lib_tests/test_files/my_test_cls.py,sha256=7ZDsFkxrLfOY6q00U5Y-daxfuhATK-K5H04RP-VmQdE,850
 h2lib_tests/test_files/DTU_10_MW/control/dtu_we_controller.dll,sha256=C5T_CuAFtIuDgCXSYAoNu24yKPwj2nWOeORacJbLN9s,1134592
@@ -87,7 +88,7 @@ h2lib_tests/test_files/minimal/res/minimal_mann_turb.hdf5,sha256=Q3cs3bZyplZjBpo
 h2lib_tests/test_files/minimal/turb/hawc2_mann_l33.6_ae0.1000_g3.9_h0_512xd32xd16_2.000x3.00x4.00_s0001_u,sha256=byiorJmXDL6uKFbyfXthHTjJdm6ELvLR2lS202KrhRI,1048576
 h2lib_tests/test_files/minimal/turb/hawc2_mann_l33.6_ae0.1000_g3.9_h0_512xd32xd16_2.000x3.00x4.00_s0001_v,sha256=cxK5Rfgfm3gyJsEYi_KlmYY8DIIl_G0aizN2jt18Glc,1048576
 h2lib_tests/test_files/minimal/turb/hawc2_mann_l33.6_ae0.1000_g3.9_h0_512xd32xd16_2.000x3.00x4.00_s0001_w,sha256=xs61jAwhP3fIR1P5Oa8ovEt2baLoF8uCNs6pKIT8L4o,1048576
-h2lib_tests-13.1.506.dist-info/METADATA,sha256=wvzBrJm0RO6QivFp3_Hx4Pol3PqGCk_9WyxwgOtyuao,349
-h2lib_tests-13.1.506.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-h2lib_tests-13.1.506.dist-info/top_level.txt,sha256=WufAL3LO35YJBhWg1AfgTjSld-6l_WuRkXAkNKczUrM,12
-h2lib_tests-13.1.506.dist-info/RECORD,,
+h2lib_tests-13.1.901.dist-info/METADATA,sha256=TUVzzgIneiYy2rlkF4hIDhuRxNBj9oBgdh25r7GvLLA,360
+h2lib_tests-13.1.901.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+h2lib_tests-13.1.901.dist-info/top_level.txt,sha256=WufAL3LO35YJBhWg1AfgTjSld-6l_WuRkXAkNKczUrM,12
+h2lib_tests-13.1.901.dist-info/RECORD,,

{h2lib_tests-13.1.506.dist-info → h2lib_tests-13.1.901.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

h2lib_tests/test_write_htc.py

@@ -1,126 +0,0 @@
-import pytest
-from wetb.hawc2.htc_file import HTCFile
-from h2lib_tests.test_files import tfp
-
-
-@pytest.fixture(scope="module")
-def write_dtu10mw_only_tower():
-    # Start from DTU_10MW_RWT and delete everything except the tower.
-    htc = HTCFile(tfp + "DTU_10_MW/htc/DTU_10MW_RWT.htc")
-    htc.set_name("DTU_10MW_RWT_only_tower")
-    for key1 in htc["new_htc_structure"].keys():
-        if key1.startswith("main_body"):
-            if "tower" not in htc["new_htc_structure"][key1]["name"].values:
-                htc["new_htc_structure"][key1].delete()
-        if key1 == "orientation":
-            for key2 in htc["new_htc_structure"]["orientation"].keys():
-                if key2.startswith("relative"):
-                    htc["new_htc_structure"]["orientation"][key2].delete()
-        if key1 == "constraint":
-            for key2 in htc["new_htc_structure"]["constraint"].keys():
-                if key2 != "fix0":
-                    htc["new_htc_structure"]["constraint"][key2].delete()
-    htc["wind"].delete()
-    htc["aerodrag"].delete()
-    htc["aero"].delete()
-    htc["dll"].delete()
-    htc["output"].delete()
-    # Reduce simulation time.
-    htc.simulation.time_stop = 10.0
-    # Change number of bodies in the tower.
-    htc.new_htc_structure.main_body.nbodies = 3
-    # Save the new file.
-    htc.save()
-    return htc
-
-
-@pytest.fixture(scope="module")
-def write_dtu10mw_only_tower_rotated(write_dtu10mw_only_tower):
-    # Start from the DTU_10MW_RWT_only_tower and rotate the tower.
-    htc = write_dtu10mw_only_tower.copy()
-    htc.set_name("DTU_10MW_RWT_only_tower_rotated")
-    alpha = 30.0
-    htc.new_htc_structure.orientation.base.body_eulerang = [
-        alpha,
-        0.0,
-        0.0,
-    ]
-    htc.save()
-    return (htc, alpha)
-
-
-@pytest.fixture(scope="module")
-def write_dtu10mw_only_tower_encrypted(write_dtu10mw_only_tower):
-    # Start from the DTU_10MW_RWT_only_tower and then encrypt the tower.
-    htc = write_dtu10mw_only_tower.copy()
-    htc.set_name("DTU_10MW_RWT_only_tower_encrypted")
-    # Only the tower is left.
-    htc.new_htc_structure.main_body.timoschenko_input.filename = "./data/DTU_10MW_RWT_Tower_st.dat.v3.enc"
-    htc.save()
-
-
-@pytest.fixture(scope="module")
-def write_dtu10mw_only_blade():
-    # Start from DTU_10MW_RWT and delete everything except the blade.
-    htc = HTCFile(tfp + "DTU_10_MW/htc/DTU_10MW_RWT.htc")
-    htc.set_name("DTU_10MW_RWT_only_blade")
-    for key1 in htc["new_htc_structure"].keys():
-        if key1.startswith("main_body"):
-            if "blade1" not in htc["new_htc_structure"][key1]["name"].values:
-                htc["new_htc_structure"][key1].delete()
-        if key1 == "orientation":
-            htc["new_htc_structure"][key1].delete()
-        if key1 == "constraint":
-            htc["new_htc_structure"][key1].delete()
-    htc["wind"].delete()
-    htc["aerodrag"].delete()
-    htc["aero"].delete()
-    htc["dll"].delete()
-    htc["output"].delete()
-
-    # Set the blade horizontal, to maximize gravity loading.
-    htc.new_htc_structure.add_section("orientation")
-    htc.new_htc_structure.orientation.add_section("base")
-    htc.new_htc_structure.orientation.base.mbdy = "blade1"
-    htc.new_htc_structure.orientation.base.inipos = [0.0, 0.0, 0.0]
-    htc.new_htc_structure.orientation.base["mbdy_eulerang"] = [90.0, 0.0, 0.0]
-    htc.new_htc_structure.orientation.base.mbdy_eulerang.comments = "Blade span is horizontal."
-
-    # Clamp the blade.
-    htc.new_htc_structure.add_section("constraint")
-    htc.new_htc_structure.constraint.add_section("fix0")
-    htc.new_htc_structure.constraint.fix0.mbdy = "blade1"
-
-    # Set as many bodies as elements.
-    htc.new_htc_structure.main_body__7.nbodies = 26
-
-    # Reduce simulation time to 1 time step.
-    htc.simulation.time_stop = 0.01
-    htc.simulation.log_deltat.delete()
-
-    # Do not use static solver, since it will be done during the test.
-    htc.simulation.solvertype = 2
-    htc.simulation.solvertype.comments = ""
-    htc.simulation.initial_condition = 1
-
-    # No output, as we will use add_sensor().
-    # htc.add_section("output")
-    # htc.output.data_format = "gtsdf"
-    # htc.output.buffer = 10000
-
-    # Save the new file.
-    htc.save()
-
-    return htc
-
-
-@pytest.fixture(scope="module")
-def write_dtu10mw_only_blade_low_max_iter(write_dtu10mw_only_blade):
-    # Start from the write_dtu10mw_only_blade and thenreduce the number of max iterations,
-    # so that the static solver will not have time to converge.
-    htc = write_dtu10mw_only_blade.copy()
-    htc.set_name("DTU_10MW_RWT_only_blade_low_max_iter")
-    htc.simulation.max_iterations = 1
-    htc.save()
-
-    return htc