h2lib-tests 13.2.702__py3-none-any.whl → 13.2.901__py3-none-any.whl

h2lib_tests/conftest.py

@@ -4,12 +4,15 @@ from wetb.hawc2.htc_file import HTCFile
 from wetb.hawc2.mainbody import MainBody
 from h2lib_tests.test_files import tfp
 from h2lib._h2lib import H2Lib
+from hawc2models import IEA22MW
 
 
+# %% Write HAWC2 models of the DTU 10 MW.
+
 @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 = HTCFile(f"{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"):
@@ -69,7 +72,7 @@ def write_dtu10mw_only_tower_encrypted(write_dtu10mw_only_tower):
 @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 = HTCFile(f"{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"):
@@ -177,7 +180,7 @@ 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 = HTCFile(f"{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
@@ -265,7 +268,7 @@ def write_dtu10mw_only_blade_rotate_relative():
 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 = HTCFile(f"{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
@@ -376,11 +379,12 @@ def write_dtu10mw_only_blade_high_gravity(
     htc.new_htc_structure.main_body__7.gravity = 100.0
     mb = MainBody(htc, "blade1")
     blade_c2def = mb.c2def
+    blade_st = mb.stFile
     # Somehow the wind and output blocks are back.
     htc["wind"].delete()
     htc["output"].delete()
     htc.save()
-    return htc, blade_c2def
+    return htc, blade_c2def, blade_st
 
 
 @pytest.fixture(scope="session")
@@ -388,17 +392,18 @@ def write_dtu10mw_only_blade_high_gravity_1_body(
     write_dtu10mw_only_blade_high_gravity,
 ):
     # Start from the write_dtu10mw_only_blade_high_gravity and then set 1 body.
-    htc_ori, _ = write_dtu10mw_only_blade_high_gravity
+    htc_ori, _, _ = write_dtu10mw_only_blade_high_gravity
     htc = htc_ori.copy()
     htc.set_name("DTU_10MW_RWT_only_blade_high_gravity_1_body")
     htc.new_htc_structure.main_body__7.nbodies = 1
     mb = MainBody(htc, "blade1")
     blade_c2def = mb.c2def
+    blade_st = mb.stFile
     # Somehow the wind and output blocks are back.
     htc["wind"].delete()
     htc["output"].delete()
     htc.save()
-    return htc, blade_c2def
+    return htc, blade_c2def, blade_st
 
 
 @pytest.fixture(scope="session")
@@ -429,6 +434,32 @@ def write_dtu10mw_only_blade_high_gravity_deformed(
     return htc, blade_c2def
 
 
+@pytest.fixture(scope="session")
+def write_dtu10mw_only_blade_high_gravity_changed_st(
+    write_dtu10mw_only_blade,
+):
+    # Start from the write_dtu10mw_only_blade and then increase the gravity loading.
+    htc = write_dtu10mw_only_blade.copy()
+    htc.set_name("DTU_10MW_RWT_only_blade_high_gravity_changed_st")
+    htc.new_htc_structure.main_body__7.gravity = 100.0
+    # Change ST
+    mb = MainBody(htc, "blade1")
+    blade_st = mb.stFile
+    rng = np.random.default_rng(seed=582)
+    factor = rng.uniform(low=0.7, high=1.3, size=19)
+    blade_st.main_data_sets[1][1] *= factor
+    # Current directory is hawc2lib/tests/h2lib_tests
+    blade_st.save(f"{tfp}DTU_10_MW/data/changed_st.dat", precision="%28.16e")
+    htc.new_htc_structure.main_body__7.timoschenko_input.filename = "./data/changed_st.dat"
+    # Somehow the wind and output blocks are back.
+    htc["wind"].delete()
+    htc["output"].delete()
+    htc.save()
+    return htc, blade_st
+
+
+# %% Create H2Lib objects of the DTU 10 MW.
+
 @pytest.fixture(scope="session")
 def h2_dtu_10mw_only_tower(write_dtu10mw_only_tower):
     h2 = H2Lib(suppress_output=True)
@@ -592,3 +623,177 @@ def h2_dtu10mw_only_blade_high_gravity_1_body(write_dtu10mw_only_blade_high_grav
     h2.stop_on_error(False)
     yield h2
     h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_dtu10mw_only_blade_high_gravity_changed_st(write_dtu10mw_only_blade_high_gravity_changed_st):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}DTU_10_MW/"
+    htc_path = "htc/DTU_10MW_RWT_only_blade_high_gravity_changed_st.htc"
+    h2.add_sensor("mbdy statevec_new blade1 c2def global absolute 90.0 1.0 0.0 0.0")  # 1-6
+    h2.add_sensor("mbdy forcevec blade1 1 1 blade1")  # 7, 8, 9
+    h2.add_sensor("mbdy momentvec blade1 1 1 blade1")  # 10, 11, 12
+    h2.init(htc_path=htc_path, model_path=model_path)
+    h2.stop_on_error(False)
+    yield h2
+    h2.close()
+
+
+# %% Write HAWC2 models of the IEA 22 MW.
+
+@pytest.fixture(scope="session")
+def write_iea22mw_rwt():
+    # Download HAWC2 model from GitHub repo.
+    htc = IEA22MW(folder=f"{tfp}IEA-22-280-RWT",
+                  version="5f9bc425bcbda0723a1245bb4e0c1fc5d5439ed3")  # Last commit on master from Aug 26, 2025
+    htc.make_onshore()
+    htc.save()
+    return htc
+
+
+@pytest.fixture(scope="session")
+def write_iea22mw_only_blade(write_iea22mw_rwt):
+    # Start from IEA-22-280-RWT and delete everything except the blade.
+    # htc = write_iea22mw_rwt.copy()
+    htc = HTCFile(f"{tfp}IEA-22-280-RWT/htc/iea_22mw_rwt.htc")
+    htc.set_name("IEA_22MW_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 low convergence limits.
+    htc.simulation.convergence_limits = [1e2, 1e-5, 1e-07]
+
+    # No output, as we will use add_sensor().
+
+    # Save the new file.
+    htc.save()
+
+    return htc
+
+
+@pytest.fixture(scope="session")
+def write_iea22mw_only_blade_high_gravity(write_iea22mw_only_blade):
+    # Start from the write_iea22mw_only_blade and then increase the gravity loading.
+    htc = write_iea22mw_only_blade.copy()
+    htc.set_name("IEA_22MW_RWT_only_blade_high_gravity")
+    htc.new_htc_structure.main_body__8.gravity = 100.0
+    mb = MainBody(htc, "blade1")
+    blade_c2def = mb.c2def
+    blade_st = mb.stFile
+    # Somehow the wind and output blocks are back.
+    htc["wind"].delete()
+    htc["output"].delete()
+    htc.save()
+    return htc, blade_c2def, blade_st
+
+
+@pytest.fixture(scope="session")
+def write_iea22mw_only_blade_high_gravity_1_body(
+    write_iea22mw_only_blade_high_gravity,
+):
+    # Start from the write_dtu10mw_only_blade_high_gravity and then set 1 body.
+    htc_ori, _, _ = write_iea22mw_only_blade_high_gravity
+    htc = htc_ori.copy()
+    htc.set_name("IEA_22MW_RWT_only_blade_high_gravity_1_body")
+    htc.new_htc_structure.main_body__8.nbodies = 1
+    mb = MainBody(htc, "blade1")
+    blade_c2def = mb.c2def
+    blade_st = mb.stFile
+    # Somehow the wind and output blocks are back.
+    htc["wind"].delete()
+    htc["output"].delete()
+    htc.save()
+    return htc, blade_c2def, blade_st
+
+
+@pytest.fixture(scope="session")
+def write_iea22mw_only_blade_high_gravity_changed_st(
+    write_iea22mw_only_blade,
+):
+    # Start from the write_iea22mw_only_blade and then increase the gravity loading.
+    htc = write_iea22mw_only_blade.copy()
+    htc.set_name("IEA_22MW_RWT_only_blade_high_gravity_changed_st")
+    htc.new_htc_structure.main_body__8.gravity = 100.0
+    # Change ST
+    mb = MainBody(htc, "blade1")
+    blade_st = mb.stFile
+    rng = np.random.default_rng(seed=582)
+    # We change all columns except for the curved length.
+    factor = rng.uniform(low=0.7, high=1.3, size=29)
+    blade_st.main_data_sets[1][1][:, 1:] *= factor
+    # Current directory is hawc2lib/tests/h2lib_tests
+    blade_st.save(f"{tfp}IEA-22-280-RWT/data/changed_st_fpm.dat", precision="%28.16e")
+    htc.new_htc_structure.main_body__8.timoschenko_input.filename = "./data/changed_st_fpm.dat"
+    # Somehow the wind and output blocks are back.
+    htc["wind"].delete()
+    htc["output"].delete()
+    htc.save()
+    return htc, blade_st
+
+
+# %% Create H2Lib objects of the IEA 22 MW.
+
+@pytest.fixture(scope="session")
+def h2_iea22mw_only_blade_high_gravity(write_iea22mw_only_blade_high_gravity):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}IEA-22-280-RWT/"
+    htc_path = "htc/IEA_22MW_RWT_only_blade_high_gravity.htc"
+    h2.add_sensor("mbdy statevec_new blade1 c2def global absolute 90.0 1.0 0.0 0.0")  # 1-6
+    h2.add_sensor("mbdy forcevec blade1 1 1 blade1")  # 7, 8, 9
+    h2.add_sensor("mbdy momentvec blade1 1 1 blade1")  # 10, 11, 12
+    h2.init(htc_path=htc_path, model_path=model_path)
+    h2.stop_on_error(False)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_iea22mw_only_blade_high_gravity_1_body(write_iea22mw_only_blade_high_gravity_1_body):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}IEA-22-280-RWT/"
+    htc_path = "htc/IEA_22MW_RWT_only_blade_high_gravity_1_body.htc"
+    h2.add_sensor("mbdy statevec_new blade1 c2def global absolute 90.0 1.0 0.0 0.0")
+    h2.init(htc_path=htc_path, model_path=model_path)
+    h2.stop_on_error(False)
+    yield h2
+    h2.close()
+
+
+@pytest.fixture(scope="session")
+def h2_iea22mw_only_blade_high_gravity_changed_st(write_iea22mw_only_blade_high_gravity_changed_st):
+    h2 = H2Lib(suppress_output=True)
+    model_path = f"{tfp}IEA-22-280-RWT/"
+    htc_path = "htc/IEA_22MW_RWT_only_blade_high_gravity_changed_st.htc"
+    h2.add_sensor("mbdy statevec_new blade1 c2def global absolute 90.0 1.0 0.0 0.0")  # 1-6
+    h2.add_sensor("mbdy forcevec blade1 1 1 blade1")  # 7, 8, 9
+    h2.add_sensor("mbdy momentvec blade1 1 1 blade1")  # 10, 11, 12
+    h2.init(htc_path=htc_path, model_path=model_path)
+    h2.stop_on_error(False)
+    yield h2
+    h2.close()