Merge pull request #1230 from raphaeltimbo/optimize-test-performance

Optimize slow tests to reduce test suite execution time

Author: raphaeltimbo

ross/tests/test_rotor_assembly.py

@@ -2288,194 +2288,75 @@ def test_ucs_calc(rotor8):
 
 
 def test_ucs_rotor9(rotor9):
+    # Simplified test: use num=5 instead of num=20 to reduce test time (~75% faster)
     exp_rotor_wn = np.array(
         [
             [
-                89.61923784473375,
-                120.89655743664413,
-                162.59246554800114,
-                217.42553100241312,
-                287.6459465590863,
-                372.935347725518,
-                466.2581426686537,
-                551.4076269641628,
-                613.1237259191993,
-                650.4689786203988,
-                671.2479800816767,
-                682.5353774789179,
-                688.6485929814294,
-                691.965477524608,
-                693.7688181845082,
-                694.7506707995042,
-                695.2857209934174,
-                695.577438207025,
-                695.7365318523782,
-                695.8233102639163,
+                89.61923786,
+                350.42266504,
+                662.41928664,
+                694.07268245,
+                695.82331031,
             ],
             [
-                124.8108608009496,
-                168.92618197956605,
-                228.5753022242499,
-                309.1399173060739,
-                417.7337051968829,
-                563.5536938601297,
-                757.9439326311913,
-                1013.3462956039914,
-                1338.3333635278868,
-                1716.48711727428,
-                2004.8471556137,
-                2078.8761601177744,
-                2097.725398329982,
-                2104.9250739620065,
-                2108.224809182831,
-                2109.8706285775806,
-                2110.7269388638374,
-                2111.182371668304,
-                2111.427443127732,
-                2111.5601496202053,
+                124.81086083,
+                523.02183296,
+                1891.18155397,
+                2108.7448203,
+                2111.56014972,
             ],
             [
-                976.610014941276,
-                979.8071596437271,
-                985.7435440438873,
-                996.8720841293019,
-                1018.0435385669914,
-                1059.040021915284,
-                1138.8336540231767,
-                1287.1396017447778,
-                1530.728131538211,
-                1866.6004173519655,
-                2224.740955271474,
-                2566.1521460703593,
-                2812.5568536324695,
-                2954.956819348177,
-                3030.8828351975917,
-                3070.974195701119,
-                3092.3284826481,
-                3103.806643182663,
-                3110.0148616594533,
-                3113.3854057718704,
+                976.61001489,
+                1046.09648924,
+                2047.58519431,
+                3043.40464892,
+                3113.38540495,
             ],
             [
-                2159.640747974065,
-                2159.756470270719,
-                2159.969971035258,
-                2160.366035282546,
-                2161.1082776995636,
-                2162.5260227884364,
-                2165.334410630516,
-                2171.3131180341056,
-                2186.1348008252444,
-                2238.1871377763114,
-                2490.9613952181917,
-                2978.354255657329,
-                3456.7805801535656,
-                3814.959990456543,
-                4040.908191796628,
-                4171.396315088025,
-                4244.09186723666,
-                4284.062958149884,
-                4305.937322108033,
-                4317.887004191494,
+                2159.64074905,
+                2162.07931839,
+                2321.68023491,
+                4080.81909788,
+                4317.88700623,
             ],
         ]
     )
-    ucs_results = rotor9.run_ucs()
+    ucs_results = rotor9.run_ucs(num=5)
     assert_allclose(ucs_results.wn, exp_rotor_wn)
 
     exp_rotor_wn = np.array(
         [
             [
-                89.61947064,
-                120.89741889,
-                162.5960395,
-                217.44095694,
-                287.71229812,
-                373.20676063,
-                467.22436151,
-                554.01174191,
-                618.0228653,
-                657.38198264,
-                679.5050572,
-                691.58822264,
-                698.15087843,
-                701.71684971,
-                703.65712249,
-                704.71396966,
-                705.29001687,
-                705.60412456,
-                705.77544067,
+                89.61947056,
+                350.61514002,
+                670.08583401,
+                703.98416314,
                 705.86888929,
             ],
             [
-                126.08902147,
-                170.65739193,
-                230.92107312,
-                312.32101392,
-                422.05546021,
-                569.45142419,
-                766.09122048,
-                1025.02536231,
-                1357.18701994,
-                1759.29639866,
-                2139.22396755,
-                2225.72234144,
-                2241.24852987,
-                2246.71661805,
-                2249.14967666,
-                2250.34715925,
-                2250.96610863,
-                2251.29417729,
-                2251.47039346,
-                2251.56572169,
+                126.08902134,
+                528.47472934,
+                1971.00830161,
+                2249.52916259,
+                2251.5657217,
             ],
             [
                 1006.73169597,
-                1010.04096437,
-                1016.17902002,
-                1027.66388689,
-                1049.44232444,
-                1091.39953214,
-                1172.55460038,
-                1322.87033841,
-                1570.73575288,
-                1916.3041552,
-                2265.71692139,
-                2655.54276614,
-                2969.42618151,
-                3163.00092959,
-                3269.25575144,
-                3325.84538219,
-                3356.03761948,
-                3372.26520948,
-                3381.03936768,
+                1078.17877977,
+                2100.24701415,
+                3286.9082896,
                 3385.80171304,
             ],
             [
                 2282.84557629,
-                2282.91523796,
-                2283.04371506,
-                2283.28189556,
-                2283.72771552,
-                2284.57736115,
-                2286.25338975,
-                2289.79373785,
-                2298.45136454,
-                2328.81587039,
-                2527.00408181,
-                3011.54457196,
-                3511.90570401,
-                3911.96812784,
-                4184.16765863,
-                4351.24520897,
-                4448.09334563,
-                4502.60333569,
-                4532.82984364,
+                2284.30991972,
+                2383.54778909,
+                4234.40738673,
                 4549.46314886,
             ],
         ]
     )
-    ucs_results = rotor9.run_ucs(synchronous=True)
+    ucs_results = rotor9.run_ucs(synchronous=True, num=5)
     assert_allclose(ucs_results.wn, exp_rotor_wn, rtol=1e-6)
 
 
@@ -2718,12 +2599,15 @@ def test_harmonic_response(rotor9):
 
 def test_amb_controller():
     # Test for the magnetic_bearing_controller method.
+    # Simplified: reduced simulation time from 10s to 2s (5x fewer time steps, ~80% faster)
     from ross.rotor_assembly import rotor_amb_example
 
     rotor = rotor_amb_example()
 
     speed = 1200
-    t = np.linspace(0, 10, 40001)
+    t = np.arange(
+        0, 2, 0.00025
+    )  # 2 seconds with dt=0.00025s (8000 steps vs original 40001)
     node = [27, 29]
     mass = [10, 10]
     probes = [12, 43]