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

h2lib_tests/test_topology_h2lib.py

@@ -238,23 +238,36 @@ def test_set_orientation_base_speed(h2_dtu_10mw_only_blade):
     )
 
 
+def test_set_c2_def_too_few_sections(h2_dtu_10mw_only_blade):
+    blade_id = h2_dtu_10mw_only_blade.get_mainbody_name_dict()["blade1"]
+    with pytest.raises(ValueError, match="TOO_FEW_SECTIONS_IN_C2DEF"):
+        h2_dtu_10mw_only_blade.set_c2_def(blade_id, np.zeros((1, 4)))
+
+
+def test_set_c2_def_wrong_number_of_columns(h2_dtu_10mw_only_blade):
+    blade_id = h2_dtu_10mw_only_blade.get_mainbody_name_dict()["blade1"]
+    with pytest.raises(ValueError, match="WRONG_NUMBER_OF_COLUMNS"):
+        h2_dtu_10mw_only_blade.set_c2_def(blade_id, np.zeros((2, 6)))
+
+
 def test_set_c2_def_main_body_not_found(h2_dtu_10mw_only_blade):
     with pytest.raises(ValueError, match="MAIN_BODY_NOT_FOUND"):
         h2_dtu_10mw_only_blade.set_c2_def(
-            "tower",
+            -10,
+            np.array([[0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0]]),
+        )
+    with pytest.raises(ValueError, match="MAIN_BODY_NOT_FOUND"):
+        h2_dtu_10mw_only_blade.set_c2_def(
+            123,
             np.array([[0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0]]),
         )
-
-
-def test_set_c2_def_too_few_sections(h2_dtu_10mw_only_blade):
-    with pytest.raises(ValueError, match="TOO_FEW_SECTIONS_IN_C2DEF"):
-        h2_dtu_10mw_only_blade.set_c2_def("blade1", np.zeros((1, 4)))
 
 
 def test_set_c2_def_different_nsec(h2_dtu_10mw_only_blade):
+    blade_id = h2_dtu_10mw_only_blade.get_mainbody_name_dict()["blade1"]
     with pytest.raises(ValueError, match="DIFFERENT_NSEC"):
         h2_dtu_10mw_only_blade.set_c2_def(
-            "blade1",
+            blade_id,
             np.array([[0.0, 0.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0]]),
         )
 
@@ -262,19 +275,21 @@ def test_set_c2_def_different_nsec(h2_dtu_10mw_only_blade):
 def test_set_c2_def_uniform_node_distribution(
     h2_dtu_10mw_only_blade_uniform_node_distribution,
 ):
+    blade_id = h2_dtu_10mw_only_blade_uniform_node_distribution.get_mainbody_name_dict()["blade1"]
     with pytest.raises(
         NotImplementedError,
     ):
         h2_dtu_10mw_only_blade_uniform_node_distribution.set_c2_def(
-            "blade1", np.ones((27, 4), dtype=np.float64, order="F")
+            blade_id, np.ones((27, 4), dtype=np.float64, order="F")
         )
 
 
 def test_set_c2_def_beam_too_short(h2_dtu_10mw_only_blade):
     c2def = np.zeros((27, 4), dtype=np.float64, order="F")
     c2def[:, 2] = np.linspace(0.0, 1e-9, c2def.shape[0])
+    blade_id = h2_dtu_10mw_only_blade.get_mainbody_name_dict()["blade1"]
     with pytest.raises(ValueError, match="BEAM_TOO_SHORT"):
-        h2_dtu_10mw_only_blade.set_c2_def("blade1", c2def)
+        h2_dtu_10mw_only_blade.set_c2_def(blade_id, c2def)
 
 
 def test_set_c2_def_blade_static_back_to_original(
@@ -288,8 +303,9 @@ def test_set_c2_def_blade_static_back_to_original(
     #  4. Revert it to the original one and check that the static solution matches the one from step 2.
 
     # Get the clamped DTU 10 MW blade subjected to high gravity loading.
-    _, blade_c2def = write_dtu10mw_only_blade_high_gravity
+    _, blade_c2def, _ = write_dtu10mw_only_blade_high_gravity
     h2 = h2_dtu10mw_only_blade_high_gravity
+    blade_id = h2.get_mainbody_name_dict()["blade1"]
 
     # Sensors 1 to 6 are the displacement and rotation of the blade tip.
     #   h2.get_sensor_info(1)
@@ -302,7 +318,7 @@ def test_set_c2_def_blade_static_back_to_original(
     c2def_new = blade_c2def.copy()
     for factor in (0.6, 0.7, 0.8, 0.9, 1.1, 1.2, 1.3, 1.4):
         c2def_new[:, 2] = factor * blade_c2def[:, 2]
-        h2.set_c2_def("blade1", c2def_new)
+        h2.set_c2_def(blade_id, c2def_new)
         # Since we are smoothly changing c2_def, it makes sense to start the static solver from the last converged solution.
         h2.solver_static_run(reset_structure=False)
 
@@ -314,7 +330,7 @@ def test_set_c2_def_blade_static_back_to_original(
             npt.assert_allclose(blade_tip_actual, blade_tip_desired)
 
     # Restore blade c2_def.
-    h2.set_c2_def("blade1", blade_c2def)
+    h2.set_c2_def(blade_id, blade_c2def)
     h2.solver_static_run(reset_structure=True)
     blade_tip_actual = h2.get_sensor_values((1, 2, 3, 4, 5, 6))
     npt.assert_allclose(blade_tip_actual, blade_tip_desired)
@@ -333,15 +349,16 @@ def test_set_c2_def_blade_static_deformed(
     # Set c2_def in the original blade, thus making it equivalent to the deformed one.
     # Then, solve again the static problem.
     _, c2def_deformed = write_dtu10mw_only_blade_high_gravity_deformed
-    h2_dtu10mw_only_blade_high_gravity.set_c2_def("blade1", c2def_deformed)
+    blade_id = h2_dtu10mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    h2_dtu10mw_only_blade_high_gravity.set_c2_def(blade_id, c2def_deformed)
     h2_dtu10mw_only_blade_high_gravity.solver_static_run(reset_structure=True)
     blade_tip_actual = h2_dtu10mw_only_blade_high_gravity_deformed.get_sensor_values((1, 2, 3, 4, 5, 6))
 
     npt.assert_allclose(blade_tip_actual, blade_tip_desired)
 
     # Restore c2_def.
-    _, c2def_original = write_dtu10mw_only_blade_high_gravity
-    h2_dtu10mw_only_blade_high_gravity.set_c2_def("blade1", c2def_original)
+    _, c2def_original, _ = write_dtu10mw_only_blade_high_gravity
+    h2_dtu10mw_only_blade_high_gravity.set_c2_def(blade_id, c2def_original)
 
 
 @pytest.mark.skip(reason="The system eigenanalysis cannot be called more than once.")
@@ -351,8 +368,9 @@ def test_set_c2_def_blade_eig_back_to_original(
 ):
     # Get the clamped DTU 10 MW blade.
     # We use the fixture with high gravity because it also returns c2_def.
-    _, blade_c2def = write_dtu10mw_only_blade_high_gravity
+    _, blade_c2def, _ = write_dtu10mw_only_blade_high_gravity
     h2 = h2_dtu10mw_only_blade_high_gravity
+    blade_id = h2.get_mainbody_name_dict()["blade1"]
 
     # Solve the eigenvalue problem.
     h2.structure_reset()
@@ -363,7 +381,7 @@ def test_set_c2_def_blade_eig_back_to_original(
     c2def_new = blade_c2def.copy()
     for factor in (0.6, 0.7, 0.8, 0.9, 1.1, 1.2, 1.3, 1.4):
         c2def_new[:, 2] = factor * blade_c2def[:, 2]
-        h2.set_c2_def("blade1", c2def_new)
+        h2.set_c2_def(blade_id, c2def_new)
         h2.structure_reset()
         h2.linearize()
         natural_frequencies_actual, damping_ratios_actual = h2.do_system_eigenanalysis(n_modes=6, include_damping=True)
@@ -375,7 +393,7 @@ def test_set_c2_def_blade_eig_back_to_original(
             npt.assert_allclose(damping_ratios_actual, damping_ratios_desired)
 
     # Restore blade c2_def.
-    h2.set_c2_def("blade1", blade_c2def)
+    h2.set_c2_def(blade_id, blade_c2def)
     h2.structure_reset()
     h2.linearize()
     natural_frequencies_actual, damping_ratios_actual = h2.do_system_eigenanalysis(n_modes=6, include_damping=True)
@@ -383,14 +401,15 @@ def test_set_c2_def_blade_eig_back_to_original(
     npt.assert_allclose(damping_ratios_actual, damping_ratios_desired)
 
 
-def test_set_c2_def_blade_mass_back_to_original(
+def test_set_c2_def_blade_inertia_back_to_original(
     write_dtu10mw_only_blade_high_gravity_1_body,
     h2_dtu10mw_only_blade_high_gravity_1_body,
 ):
-    # Get the clamped DTU 10 MW blade.
+    # Get the clamped DTU 10 MW blade with only 1 body.
     # We use the fixture with high gravity because it also returns c2_def.
-    _, blade_c2def = write_dtu10mw_only_blade_high_gravity_1_body
+    _, blade_c2def, _ = write_dtu10mw_only_blade_high_gravity_1_body
     h2 = h2_dtu10mw_only_blade_high_gravity_1_body
+    blade_id = h2.get_mainbody_name_dict()["blade1"]
 
     # Get the inertia properties.
     h2.structure_reset()
@@ -400,7 +419,7 @@ def test_set_c2_def_blade_mass_back_to_original(
     c2def_new = blade_c2def.copy()
     for factor in (0.6, 0.7, 0.8, 0.9, 1.1, 1.2, 1.3, 1.4):
         c2def_new[:, 2] = factor * blade_c2def[:, 2]
-        h2.set_c2_def("blade1", c2def_new)
+        h2.set_c2_def(blade_id, c2def_new)
         inertia_actual = h2.body_output_mass(0)
 
         # Must differ from the desired ones.
@@ -409,13 +428,13 @@ def test_set_c2_def_blade_mass_back_to_original(
                 npt.assert_allclose(inertia_actual[i], inertia_desired[i])
 
     # Restore blade c2_def.
-    h2.set_c2_def("blade1", blade_c2def)
+    h2.set_c2_def(blade_id, blade_c2def)
     inertia_actual = h2.body_output_mass(0)
     for i in range(4):  # Loop over tuple of arrays.
         npt.assert_allclose(inertia_actual[i], inertia_desired[i])
 
 
-def test_set_c2_def_blade_mass_deformed(
+def test_set_c2_def_blade_inertia_deformed(
     write_dtu10mw_only_blade_high_gravity,
     write_dtu10mw_only_blade_high_gravity_deformed,
     h2_dtu10mw_only_blade_high_gravity,
@@ -427,7 +446,8 @@ def test_set_c2_def_blade_mass_deformed(
 
     # Set c2_def in the original blade, thus making it equivalent to the deformed one.
     _, c2def_deformed = write_dtu10mw_only_blade_high_gravity_deformed
-    h2_dtu10mw_only_blade_high_gravity.set_c2_def("blade1", c2def_deformed)
+    blade_id = h2_dtu10mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    h2_dtu10mw_only_blade_high_gravity.set_c2_def(blade_id, c2def_deformed)
 
     # Compare inertia properties for all bodies.
     nbdy, _ = h2_dtu10mw_only_blade_high_gravity_deformed.get_number_of_bodies_and_constraints()
@@ -439,8 +459,8 @@ def test_set_c2_def_blade_mass_deformed(
             npt.assert_allclose(inertia_actual[i], inertia_desired[i], rtol=1e-6)
 
     # Restore c2_def.
-    _, c2def_original = write_dtu10mw_only_blade_high_gravity
-    h2_dtu10mw_only_blade_high_gravity.set_c2_def("blade1", c2def_original)
+    _, c2def_original, _ = write_dtu10mw_only_blade_high_gravity
+    h2_dtu10mw_only_blade_high_gravity.set_c2_def(blade_id, c2def_original)
 
 
 def test_set_c2_def_blade_element_deformed(
@@ -455,7 +475,8 @@ def test_set_c2_def_blade_element_deformed(
 
     # Set c2_def in the original blade, thus making it equivalent to the deformed one.
     _, c2def_deformed = write_dtu10mw_only_blade_high_gravity_deformed
-    h2_dtu10mw_only_blade_high_gravity.set_c2_def("blade1", c2def_deformed)
+    blade_id = h2_dtu10mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    h2_dtu10mw_only_blade_high_gravity.set_c2_def(blade_id, c2def_deformed)
 
     # Compare element matrices for all bodies.
     nelem = h2_dtu10mw_only_blade_high_gravity_deformed.get_number_of_elements()
@@ -469,8 +490,377 @@ def test_set_c2_def_blade_element_deformed(
             npt.assert_allclose(mat_actual[2], mat_desired[2], rtol=1e-10)  # damping
 
     # Restore c2_def.
-    _, c2def_original = write_dtu10mw_only_blade_high_gravity
-    h2_dtu10mw_only_blade_high_gravity.set_c2_def("blade1", c2def_original)
+    _, c2def_original, _ = write_dtu10mw_only_blade_high_gravity
+    h2_dtu10mw_only_blade_high_gravity.set_c2_def(blade_id, c2def_original)
+
+
+def test_set_st_wrong_number_of_columns(h2_dtu_10mw_only_blade):
+    blade_id = h2_dtu_10mw_only_blade.get_mainbody_name_dict()["blade1"]
+    with pytest.raises(ValueError, match="WRONG_NUMBER_OF_COLUMNS"):
+        h2_dtu_10mw_only_blade.set_st(blade_id, np.empty((2, 6)))
+
+
+def test_set_st_main_body_not_found(h2_dtu_10mw_only_blade):
+    with pytest.raises(ValueError, match="MAIN_BODY_NOT_FOUND"):
+        h2_dtu_10mw_only_blade.set_st(-3, np.empty((2, 19)))
+    with pytest.raises(ValueError, match="MAIN_BODY_NOT_FOUND"):
+        h2_dtu_10mw_only_blade.set_st(123, np.empty((2, 19)))
+
+
+def test_set_st_z_not_continuously_increasing(h2_dtu_10mw_only_blade):
+    blade_id = h2_dtu_10mw_only_blade.get_mainbody_name_dict()["blade1"]
+    with pytest.raises(ValueError, match="ST_Z_NOT_CONTINUOUSLY_INCREASING"):
+        h2_dtu_10mw_only_blade.set_st(blade_id, np.zeros((3, 19)))
+
+
+def test_set_st_uniform_node_distribution(
+    h2_dtu_10mw_only_blade_uniform_node_distribution,
+):
+    blade_id = h2_dtu_10mw_only_blade_uniform_node_distribution.get_mainbody_name_dict()["blade1"]
+    with pytest.raises(NotImplementedError):
+        st = np.zeros((2, 19))
+        st[1, 0] = 1.0
+        h2_dtu_10mw_only_blade_uniform_node_distribution.set_st(blade_id, st)
+
+
+def test_set_st_classic_timoshenko_blade_static_back_to_original(
+    write_dtu10mw_only_blade_high_gravity,
+    h2_dtu10mw_only_blade_high_gravity,
+):
+    # This test will:
+    #  1. Take the DTU 10 MW blade subjected to high gravity loading.
+    #  2. Compute the static solution.
+    #  3. Change ST a few times and re-compute the static solution.
+    #  4. Revert it to the original one and check that the static solution matches the one from step 2.
+
+    # Get the clamped DTU 10 MW blade subjected to high gravity loading.
+    _, _, blade_st = write_dtu10mw_only_blade_high_gravity
+    h2 = h2_dtu10mw_only_blade_high_gravity
+    blade_id = h2.get_mainbody_name_dict()["blade1"]
+
+    # Sensors 1 to 6 are the displacement and rotation of the blade tip.
+    #   h2.get_sensor_info(1)
+
+    # Run the static solver and get the blade tip position and rotation.
+    h2.solver_static_run(reset_structure=True)
+    blade_tip_desired = h2.get_sensor_values((1, 2, 3, 4, 5, 6))
+
+    # Change blade ST and run the static solver.
+    rng = np.random.default_rng(seed=582)
+    for _ in range(10):
+        factor = rng.uniform(low=0.7, high=1.3, size=19)
+        st_new = factor * blade_st.main_data_sets[1][1]
+        # We do not change z, to prevent ST_Z_NOT_CONTINUOUSLY_INCREASING.
+        st_new[:, 0] = blade_st.main_data_sets[1][1][:, 0]
+        h2.set_st(blade_id, st_new)
+        h2.solver_static_run(reset_structure=True)
+
+        # Get new blade displacement.
+        blade_tip_actual = h2.get_sensor_values((1, 2, 3, 4, 5, 6))
+
+        # Must differ from blade_tip_desired.
+        with npt.assert_raises(AssertionError):
+            npt.assert_allclose(blade_tip_actual, blade_tip_desired)
+
+    # Restore blade c2_def.
+    h2.set_st(blade_id, blade_st.main_data_sets[1][1])
+    h2.solver_static_run(reset_structure=True)
+    blade_tip_actual = h2.get_sensor_values((1, 2, 3, 4, 5, 6))
+    npt.assert_allclose(blade_tip_actual, blade_tip_desired)
+
+
+def test_set_st_classic_timoshenko_blade_static_changed_st(
+    write_dtu10mw_only_blade_high_gravity,
+    write_dtu10mw_only_blade_high_gravity_changed_st,
+    h2_dtu10mw_only_blade_high_gravity,
+    h2_dtu10mw_only_blade_high_gravity_changed_st,
+):
+    blade_id = h2_dtu10mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    # Solve the static problem with the changed ST loaded directly by HAWC2.
+    h2_dtu10mw_only_blade_high_gravity_changed_st.solver_static_run(reset_structure=True)
+    blade_tip_desired = h2_dtu10mw_only_blade_high_gravity_changed_st.get_sensor_values((1, 2, 3, 4, 5, 6))
+
+    # Set ST in the original blade, thus making it equivalent to the changed one.
+    # Then, solve again the static problem.
+    _, st_changed = write_dtu10mw_only_blade_high_gravity_changed_st
+    h2_dtu10mw_only_blade_high_gravity.set_st(blade_id, st_changed.main_data_sets[1][1])
+    h2_dtu10mw_only_blade_high_gravity.solver_static_run(reset_structure=True)
+    blade_tip_actual = h2_dtu10mw_only_blade_high_gravity_changed_st.get_sensor_values((1, 2, 3, 4, 5, 6))
+
+    npt.assert_allclose(blade_tip_actual, blade_tip_desired)
+
+    # Restore c2_def.
+    _, _, st_riginal = write_dtu10mw_only_blade_high_gravity
+    h2_dtu10mw_only_blade_high_gravity.set_st(blade_id, st_riginal.main_data_sets[1][1])
+
+
+def test_set_st_classic_timoshenko_inertia_back_to_original(
+    write_dtu10mw_only_blade_high_gravity_1_body,
+    h2_dtu10mw_only_blade_high_gravity_1_body,
+):
+    # Get the clamped DTU 10 MW blade with only 1 body.
+    # We use the fixture with high gravity because it also returns st.
+    _, _, st = write_dtu10mw_only_blade_high_gravity_1_body
+    h2 = h2_dtu10mw_only_blade_high_gravity_1_body
+    blade_id = h2.get_mainbody_name_dict()["blade1"]
+
+    # Get the inertia properties.
+    h2.structure_reset()
+    inertia_desired = h2.body_output_mass(0)
+
+    # Change blade density and compute inertia properties.
+    mass_original = st.m()
+    rng = np.random.default_rng(seed=582)
+    for _ in range(10):
+        # Uniformly scale to change the mass.
+        density_new = rng.uniform(low=0.5, high=2.0) * mass_original
+        # Make the blade tip heavier to change center of gravity.
+        density_new *= np.linspace(0.8, 2.5, density_new.size)
+        st.set_value(mset=1, set=1, m=density_new)
+        h2.set_st(blade_id, st.main_data_sets[1][1])
+        inertia_actual = h2.body_output_mass(0)
+
+        # Must differ from the desired ones.
+        for i in range(4):  # Loop over tuple of arrays.
+            with npt.assert_raises(AssertionError):
+                npt.assert_allclose(inertia_actual[i], inertia_desired[i])
+
+    # Restore ST.
+    st.set_value(mset=1, set=1, m=mass_original)
+    h2.set_st(blade_id, st.main_data_sets[1][1])
+    inertia_actual = h2.body_output_mass(0)
+    for i in range(4):  # Loop over tuple of arrays.
+        npt.assert_allclose(inertia_actual[i], inertia_desired[i])
+
+
+def test_set_st_classic_timoshenko_blade_inertia_changed_st(
+    write_dtu10mw_only_blade_high_gravity,
+    write_dtu10mw_only_blade_high_gravity_changed_st,
+    h2_dtu10mw_only_blade_high_gravity,
+    h2_dtu10mw_only_blade_high_gravity_changed_st,
+):
+    # Revert blades to the undeflected configuration.
+    h2_dtu10mw_only_blade_high_gravity.structure_reset()
+    h2_dtu10mw_only_blade_high_gravity_changed_st.structure_reset()
+
+    # Set ST in the original blade, thus making it equivalent to the changed one.
+    _, st_changed = write_dtu10mw_only_blade_high_gravity_changed_st
+    blade_id = h2_dtu10mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    h2_dtu10mw_only_blade_high_gravity.set_st(blade_id, st_changed.main_data_sets[1][1])
+
+    # Compare inertia properties for all bodies.
+    nbdy, _ = h2_dtu10mw_only_blade_high_gravity_changed_st.get_number_of_bodies_and_constraints()
+    for i in range(nbdy):
+        inertia_desired = h2_dtu10mw_only_blade_high_gravity_changed_st.body_output_mass(i)
+        inertia_actual = h2_dtu10mw_only_blade_high_gravity.body_output_mass(i)
+
+        for i in range(4):  # Loop over tuple of arrays.
+            npt.assert_allclose(inertia_actual[i], inertia_desired[i])
+
+    # Restore ST.
+    _, _, st_original = write_dtu10mw_only_blade_high_gravity
+    h2_dtu10mw_only_blade_high_gravity.set_st(blade_id, st_original.main_data_sets[1][1])
+
+
+def test_set_st_classic_timoshenko_blade_element_changed_st(
+    write_dtu10mw_only_blade_high_gravity,
+    write_dtu10mw_only_blade_high_gravity_changed_st,
+    h2_dtu10mw_only_blade_high_gravity,
+    h2_dtu10mw_only_blade_high_gravity_changed_st,
+):
+    # Revert blades to the undeflected configuration.
+    h2_dtu10mw_only_blade_high_gravity.structure_reset()
+    h2_dtu10mw_only_blade_high_gravity_changed_st.structure_reset()
+
+    # Set ST in the original blade, thus making it equivalent to the changed one.
+    _, st_changed = write_dtu10mw_only_blade_high_gravity_changed_st
+    blade_id = h2_dtu10mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    h2_dtu10mw_only_blade_high_gravity.set_st(blade_id, st_changed.main_data_sets[1][1])
+
+    # Compare element matrices for all bodies.
+    nelem = h2_dtu10mw_only_blade_high_gravity_changed_st.get_number_of_elements()
+    for ibdy in range(nelem.size):
+        for ielem in range(nelem[ibdy]):
+            mat_desired = h2_dtu10mw_only_blade_high_gravity_changed_st.body_output_element(ibdy, ielem)
+            mat_actual = h2_dtu10mw_only_blade_high_gravity.body_output_element(ibdy, ielem)
+
+            npt.assert_allclose(mat_actual[0], mat_desired[0], rtol=1e-14)  # mass
+            npt.assert_allclose(mat_actual[1], mat_desired[1], rtol=1e-14)  # stiffness
+            npt.assert_allclose(mat_actual[2], mat_desired[2], rtol=1e-14)  # damping
+
+    # Restore ST.
+    _, _, st_original = write_dtu10mw_only_blade_high_gravity
+    h2_dtu10mw_only_blade_high_gravity.set_st(blade_id, st_original.main_data_sets[1][1])
+
+
+def test_set_st_fpm_blade_static_back_to_original(
+    write_iea22mw_only_blade_high_gravity,
+    h2_iea22mw_only_blade_high_gravity,
+):
+    # This test will:
+    #  1. Take the IEA 22 MW blade subjected to high gravity loading.
+    #  2. Compute the static solution.
+    #  3. Change ST a few times and re-compute the static solution.
+    #  4. Revert it to the original one and check that the static solution matches the one from step 2.
+
+    # Get the clamped IEA 22 MW blade subjected to high gravity loading.
+    _, _, blade_st = write_iea22mw_only_blade_high_gravity
+    h2 = h2_iea22mw_only_blade_high_gravity
+    blade_id = h2.get_mainbody_name_dict()["blade1"]
+
+    # Sensors 1 to 6 are the displacement and rotation of the blade tip.
+    #   h2.get_sensor_info(1)
+
+    # Run the static solver and get the blade tip position and rotation.
+    h2.solver_static_run(reset_structure=True)
+    blade_tip_desired = h2.get_sensor_values((1, 2, 3, 4, 5, 6))
+
+    # Change blade ST and run the static solver.
+    rng = np.random.default_rng(seed=582)
+    for _ in range(10):
+        factor = rng.uniform(low=0.7, high=1.3, size=30)
+        st_new = factor * blade_st.main_data_sets[1][1]
+        # We do not change z, to prevent ST_Z_NOT_CONTINUOUSLY_INCREASING.
+        st_new[:, 0] = blade_st.main_data_sets[1][1][:, 0]
+        h2.set_st(blade_id, st_new)
+        h2.solver_static_run(reset_structure=True)
+
+        # Get new blade displacement.
+        blade_tip_actual = h2.get_sensor_values((1, 2, 3, 4, 5, 6))
+
+        # Must differ from blade_tip_desired.
+        with npt.assert_raises(AssertionError):
+            npt.assert_allclose(blade_tip_actual, blade_tip_desired)
+
+    # Restore blade c2_def.
+    h2.set_st(blade_id, blade_st.main_data_sets[1][1])
+    h2.solver_static_run(reset_structure=True)
+    blade_tip_actual = h2.get_sensor_values((1, 2, 3, 4, 5, 6))
+    npt.assert_allclose(blade_tip_actual, blade_tip_desired)
+
+
+def test_set_st_fpm_blade_static_changed_st(
+    write_iea22mw_only_blade_high_gravity,
+    write_iea22mw_only_blade_high_gravity_changed_st,
+    h2_iea22mw_only_blade_high_gravity,
+    h2_iea22mw_only_blade_high_gravity_changed_st,
+):
+    # Solve the static problem with the changed ST loaded directly by HAWC2.
+    h2_iea22mw_only_blade_high_gravity_changed_st.solver_static_run(reset_structure=True)
+    blade_tip_desired = h2_iea22mw_only_blade_high_gravity_changed_st.get_sensor_values((1, 2, 3, 4, 5, 6))
+
+    # Set ST in the original blade, thus making it equivalent to the changed one.
+    # Then, solve again the static problem.
+    _, st_changed = write_iea22mw_only_blade_high_gravity_changed_st
+    blade_id = h2_iea22mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    h2_iea22mw_only_blade_high_gravity.set_st(blade_id, st_changed.main_data_sets[1][1])
+    h2_iea22mw_only_blade_high_gravity.solver_static_run(reset_structure=True)
+    blade_tip_actual = h2_iea22mw_only_blade_high_gravity_changed_st.get_sensor_values((1, 2, 3, 4, 5, 6))
+
+    npt.assert_allclose(blade_tip_actual, blade_tip_desired)
+
+    # Restore c2_def.
+    _, _, st_riginal = write_iea22mw_only_blade_high_gravity
+    h2_iea22mw_only_blade_high_gravity.set_st(blade_id, st_riginal.main_data_sets[1][1])
+
+
+def test_set_st_fpm_inertia_back_to_original(
+    write_iea22mw_only_blade_high_gravity_1_body,
+    h2_iea22mw_only_blade_high_gravity_1_body,
+):
+    # Get the clamped DTU 10 MW blade with only 1 body.
+    # We use the fixture with high gravity because it also returns st.
+    _, _, st = write_iea22mw_only_blade_high_gravity_1_body
+    h2 = h2_iea22mw_only_blade_high_gravity_1_body
+    blade_id = h2.get_mainbody_name_dict()["blade1"]
+
+    # Get the inertia properties.
+    h2.structure_reset()
+    inertia_desired = h2.body_output_mass(0)
+
+    # Change blade density and compute inertia properties.
+    mass_original = st.m()
+    rng = np.random.default_rng(seed=582)
+    for _ in range(10):
+        # Uniformly scale to change the mass.
+        density_new = rng.uniform(low=0.5, high=2.0) * mass_original
+        # Make the blade tip heavier to change center of gravity.
+        density_new *= np.linspace(0.8, 2.5, density_new.size)
+        st.set_value(mset=1, set=1, m=density_new)
+        h2.set_st(blade_id, st.main_data_sets[1][1])
+        inertia_actual = h2.body_output_mass(0)
+
+        # Must differ from the desired ones.
+        for i in range(4):  # Loop over tuple of arrays.
+            with npt.assert_raises(AssertionError):
+                npt.assert_allclose(inertia_actual[i], inertia_desired[i])
+
+    # Restore ST.
+    st.set_value(mset=1, set=1, m=mass_original)
+    h2.set_st(blade_id, st.main_data_sets[1][1])
+    inertia_actual = h2.body_output_mass(0)
+    for i in range(4):  # Loop over tuple of arrays.
+        npt.assert_allclose(inertia_actual[i], inertia_desired[i])
+
+
+def test_set_st_fpm_blade_inertia_changed_st(
+    write_iea22mw_only_blade_high_gravity,
+    write_iea22mw_only_blade_high_gravity_changed_st,
+    h2_iea22mw_only_blade_high_gravity,
+    h2_iea22mw_only_blade_high_gravity_changed_st,
+):
+    # Revert blades to the undeflected configuration.
+    h2_iea22mw_only_blade_high_gravity.structure_reset()
+    h2_iea22mw_only_blade_high_gravity_changed_st.structure_reset()
+
+    # Set ST in the original blade, thus making it equivalent to the changed one.
+    _, st_changed = write_iea22mw_only_blade_high_gravity_changed_st
+    blade_id = h2_iea22mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    h2_iea22mw_only_blade_high_gravity.set_st(blade_id, st_changed.main_data_sets[1][1])
+
+    # Compare inertia properties for all bodies.
+    nbdy, _ = h2_iea22mw_only_blade_high_gravity_changed_st.get_number_of_bodies_and_constraints()
+    for i in range(nbdy):
+        inertia_desired = h2_iea22mw_only_blade_high_gravity_changed_st.body_output_mass(i)
+        inertia_actual = h2_iea22mw_only_blade_high_gravity.body_output_mass(i)
+
+        for i in range(4):  # Loop over tuple of arrays.
+            npt.assert_allclose(inertia_actual[i], inertia_desired[i])
+
+    # Restore ST.
+    _, _, st_original = write_iea22mw_only_blade_high_gravity
+    h2_iea22mw_only_blade_high_gravity.set_st(blade_id, st_original.main_data_sets[1][1])
+
+
+def test_set_st_fpm_blade_element_changed_st(
+    write_iea22mw_only_blade_high_gravity,
+    write_iea22mw_only_blade_high_gravity_changed_st,
+    h2_iea22mw_only_blade_high_gravity,
+    h2_iea22mw_only_blade_high_gravity_changed_st,
+):
+    # Revert blades to the undeflected configuration.
+    h2_iea22mw_only_blade_high_gravity.structure_reset()
+    h2_iea22mw_only_blade_high_gravity_changed_st.structure_reset()
+
+    # Set ST in the original blade, thus making it equivalent to the changed one.
+    _, st_changed = write_iea22mw_only_blade_high_gravity_changed_st
+    blade_id = h2_iea22mw_only_blade_high_gravity.get_mainbody_name_dict()["blade1"]
+    h2_iea22mw_only_blade_high_gravity.set_st(blade_id, st_changed.main_data_sets[1][1])
+
+    # Compare element matrices for all bodies.
+    nelem = h2_iea22mw_only_blade_high_gravity_changed_st.get_number_of_elements()
+    for ibdy in range(nelem.size):
+        for ielem in range(nelem[ibdy]):
+            mat_desired = h2_iea22mw_only_blade_high_gravity_changed_st.body_output_element(ibdy, ielem)
+            mat_actual = h2_iea22mw_only_blade_high_gravity.body_output_element(ibdy, ielem)
+
+            npt.assert_allclose(mat_actual[0], mat_desired[0], rtol=1e-14)  # mass
+            npt.assert_allclose(mat_actual[1], mat_desired[1], rtol=1e-14)  # stiffness
+            npt.assert_allclose(mat_actual[2], mat_desired[2], rtol=1e-14)  # damping
+
+    # Restore ST.
+    _, _, st_original = write_iea22mw_only_blade_high_gravity
+    h2_iea22mw_only_blade_high_gravity.set_st(blade_id, st_original.main_data_sets[1][1])
 
 
 def test_set_orientation_relative_main_body_not_found(
@@ -605,5 +995,5 @@ def test_set_orientation_relative_static(
 # %% Main.
 
 if __name__ == "__main__":
-    # pytest.main([__file__])
-    pytest.main([__file__, "-k test_set_c2_def_blade_element_deformed"])
+    pytest.main([__file__])
+    # pytest.main([__file__, "-k test_set_st_fpm_blade_element_changed_st"])

{h2lib_tests-13.2.702.dist-info → h2lib_tests-13.2.901.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: h2lib-tests
-Version: 13.2.702
-Summary: Tests and test_files for test h2lib (13.2.7+4-gb779ee7)
+Version: 13.2.901
+Summary: Tests and test_files for test h2lib (13.2.9+4-gfd77867)
 Download-URL: 
 Author: Mads M Pedersen
 Author-email: 
@@ -13,6 +13,7 @@ Requires-Dist: pytest-cov
 Requires-Dist: hipersim
 Requires-Dist: wetb
 Requires-Dist: h2lib
+Requires-Dist: Hawc2Models==0.0.4
 Dynamic: author
 Dynamic: requires-dist
 Dynamic: summary

{h2lib_tests-13.2.702.dist-info → h2lib_tests-13.2.901.dist-info}/RECORD

@@ -1,16 +1,16 @@
 h2lib_tests/__init__.py,sha256=VjSqfGg8BzdmSjfSFhJh4hZbYZ_cME7xp9EWFKHQphA,61
-h2lib_tests/conftest.py,sha256=o9bkMEaB40rr_KSGovy1nuvq2Y09rKfQWxTcTGZ0MfY,21607
+h2lib_tests/conftest.py,sha256=r4p868zCfJ5MMd2hs6XvGaiqu3jYlSR5G0OKAk5WX7Y,29629
 h2lib_tests/dtu10mw.py,sha256=a7SXfyDwDQPastYKb5CgghOQcYfgO1eTwGrd-H3Enok,4374
 h2lib_tests/test_calc.py,sha256=VNLfr2J9R2Jy9xTbdZ9dfbQ4dCwr7H7nbZRI3yP69fQ,2152
-h2lib_tests/test_distributed_sections.py,sha256=JtguIddd2nmRyNsy2jqjPFB8E5cGpH2_IVErP1wGPPg,5432
+h2lib_tests/test_distributed_sections.py,sha256=xYcQwNGL3IQyVaguHGZrm0EZGYIMw4LRq9q5xwird1E,19413
 h2lib_tests/test_ellipsys_couplings.py,sha256=mfSTk1IamaKETU6Be8p8W9vA-yostWJl17m5sFlEK3Q,3345
-h2lib_tests/test_h2lib.py,sha256=LEV1n0Fwa1llQwIM06Mcqr07Y6WOn-j8mNxjzlCEOl0,14219
-h2lib_tests/test_h2rotor.py,sha256=yAcG9RDD3uO4g4w8LJnJ3Md_bC-_mBn1--4KHwvlTRc,12963
+h2lib_tests/test_h2lib.py,sha256=t2jwmXzibej9DQoAkldwWX1MmEmpcV0kp2hLDRV5dbQ,13836
+h2lib_tests/test_h2rotor.py,sha256=cF3KMdzTyGoZGyihhHCJqdH_n_t2vjS1BPHDaUCk9pU,13918
 h2lib_tests/test_lin.py,sha256=yydbsMX44ym_MSqR1cbQkqil8qC4icSW_JccPSElGac,6292
 h2lib_tests/test_mpi.py,sha256=CTT160yc6uZFRr_QNFWxyOwJ-y0qHiIp8jPYs1MQpyg,7111
 h2lib_tests/test_multiprocessinterface.py,sha256=h2o4havtK6IuMXsplNjGUa3VxOnbpEYGxdrrAKQilj0,1470
 h2lib_tests/test_static_solver.py,sha256=IuUkKrEoz_EKLCmGQ94CLFgjKPvO20iVx4773JTKO-8,6503
-h2lib_tests/test_topology_h2lib.py,sha256=7ZfVjEUYBJacPcFCDi_Qb0LqW6W-0no_zTBUKbNcKQw,23858
+h2lib_tests/test_topology_h2lib.py,sha256=8SCrrjd8VZh_1pX9KDm_b2aCCyCZ-YbjwRgBRCOKNwQ,42472
 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
@@ -91,7 +91,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.2.702.dist-info/METADATA,sha256=Bm3FrFz3BYvqhQnv-UOFNbfIkg1fbgEcP8KAc9QYMtk,417
-h2lib_tests-13.2.702.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-h2lib_tests-13.2.702.dist-info/top_level.txt,sha256=WufAL3LO35YJBhWg1AfgTjSld-6l_WuRkXAkNKczUrM,12
-h2lib_tests-13.2.702.dist-info/RECORD,,
+h2lib_tests-13.2.901.dist-info/METADATA,sha256=tsWn93wEcdivKQ20oO9utThtj-uQRsQWDZmGR_khWj0,452
+h2lib_tests-13.2.901.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+h2lib_tests-13.2.901.dist-info/top_level.txt,sha256=WufAL3LO35YJBhWg1AfgTjSld-6l_WuRkXAkNKczUrM,12
+h2lib_tests-13.2.901.dist-info/RECORD,,