foxes 1.1.1__py3-none-any.whl → 1.2__py3-none-any.whl

tests/1_verification/flappy_0_6/row_Jensen_linear_centre/test_row_Jensen_linear_centre.py

@@ -5,7 +5,6 @@ import inspect
 
 import foxes
 import foxes.variables as FV
-import foxes.constants as FC
 
 thisdir = Path(inspect.getfile(inspect.currentframe())).parent
 
@@ -13,7 +12,6 @@ thisdir = Path(inspect.getfile(inspect.currentframe())).parent
 def test():
     n_s = 800
     n_t = 76
-    c = 1000
     p0 = np.array([0.0, 0.0])
     stp = np.array([497.0, 0.0])
     cfile = thisdir / "flappy" / "results.csv.gz"
@@ -39,54 +37,56 @@ def test():
         verbosity=0,
     )
 
-    algo = foxes.algorithms.Downwind(
-        farm,
-        states,
-        mbook=mbook,
-        rotor_model="centre",
-        wake_models=["Jensen_linear_k007"],
-        wake_frame="rotor_wd",
-        partial_wakes={"Jensen_linear_k007": "rotor_points"},
-        verbosity=0,
-    )
-
-    data = algo.calc_farm()
-
-    df = data.to_dataframe()[[FV.WD, FV.AMB_REWS, FV.REWS, FV.AMB_P, FV.P]]
-
-    print("\nReading file", cfile)
-    fdata = pd.read_csv(cfile).set_index(["state", "turbine"])
-
-    print()
-    print("TRESULTS\n")
-    sel = (df[FV.P] > 0) & (fdata[FV.P] > 0)
-    df = df.loc[sel]
-    fdata = fdata.loc[sel]
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-
-    print("\nVERIFYING\n")
-    df[FV.WS] = df["REWS"]
-    df[FV.AMB_WS] = df["AMB_REWS"]
-
-    delta = df - fdata
-    print(delta)
-
-    chk = delta.abs()
-    print(chk.max())
-
-    var = FV.WS
-    print(f"\nCHECKING {var}")
-    sel = chk[var] >= 1e-7
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-    print(chk.loc[sel])
-    assert (chk[var] < 1e-7).all()
-
-    var = FV.P
-    sel = chk[var] >= 1e-5
-    print(f"\nCHECKING {var}\n", delta.loc[sel])
-    assert (chk[var] < 1e-5).all()
+    with foxes.Engine.new("threads", chunk_size_states=1000):
+
+        algo = foxes.algorithms.Downwind(
+            farm,
+            states,
+            mbook=mbook,
+            rotor_model="centre",
+            wake_models=["Jensen_linear_k007"],
+            wake_frame="rotor_wd",
+            partial_wakes={"Jensen_linear_k007": "rotor_points"},
+            verbosity=0,
+        )
+
+        data = algo.calc_farm()
+
+        df = data.to_dataframe()[[FV.WD, FV.AMB_REWS, FV.REWS, FV.AMB_P, FV.P]]
+
+        print("\nReading file", cfile)
+        fdata = pd.read_csv(cfile).set_index(["state", "turbine"])
+
+        print()
+        print("TRESULTS\n")
+        sel = (df[FV.P] > 0) & (fdata[FV.P] > 0)
+        df = df.loc[sel]
+        fdata = fdata.loc[sel]
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+
+        print("\nVERIFYING\n")
+        df[FV.WS] = df["REWS"]
+        df[FV.AMB_WS] = df["AMB_REWS"]
+
+        delta = df - fdata
+        print(delta)
+
+        chk = delta.abs()
+        print(chk.max())
+
+        var = FV.WS
+        print(f"\nCHECKING {var}")
+        sel = chk[var] >= 1e-7
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+        print(chk.loc[sel])
+        assert (chk[var] < 1e-7).all()
+
+        var = FV.P
+        sel = chk[var] >= 1e-5
+        print(f"\nCHECKING {var}\n", delta.loc[sel])
+        assert (chk[var] < 1e-5).all()
 
 
 if __name__ == "__main__":

tests/1_verification/flappy_0_6/row_Jensen_linear_tophat/test_row_Jensen_linear_tophat.py

@@ -5,7 +5,7 @@ import inspect
 
 import foxes
 import foxes.variables as FV
-import foxes.constants as FC
+from foxes.config import config
 
 thisdir = Path(inspect.getfile(inspect.currentframe())).parent
 
@@ -19,8 +19,6 @@ def test():
     cfile = thisdir / "flappy" / "results.csv.gz"
     tfile = thisdir / "NREL-5MW-D126-H90.csv"
 
-    ck = {FC.STATE: c}
-
     mbook = foxes.models.ModelBook()
     ttype = foxes.models.turbine_types.PCtFile(
         data_source=tfile, var_ws_ct=FV.REWS, var_ws_P=FV.REWS
@@ -41,55 +39,56 @@ def test():
         verbosity=0,
     )
 
-    algo = foxes.algorithms.Downwind(
-        farm,
-        states,
-        mbook=mbook,
-        rotor_model="centre",
-        wake_models=["Jensen_linear_k007"],
-        wake_frame="rotor_wd",
-        partial_wakes={"Jensen_linear_k007": "top_hat"},
-        chunks=ck,
-        verbosity=0,
-    )
-
-    data = algo.calc_farm()
-
-    df = data.to_dataframe()[[FV.WD, FV.AMB_REWS, FV.REWS, FV.AMB_P, FV.P]]
-
-    print("\nReading file", cfile)
-    fdata = pd.read_csv(cfile).set_index(["state", "turbine"])
-
-    print()
-    print("TRESULTS\n")
-    sel = (df[FV.P] > 0) & (fdata[FV.P] > 0)
-    df = df.loc[sel]
-    fdata = fdata.loc[sel]
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-
-    print("\nVERIFYING\n")
-    df[FV.WS] = df["REWS"]
-    df[FV.AMB_WS] = df["AMB_REWS"]
-
-    delta = df - fdata
-    print(delta)
-
-    chk = delta.abs()
-    print(chk.max())
-
-    var = FV.WS
-    print(f"\nCHECKING {var}")
-    sel = chk[var] >= 1e-7
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-    print(chk.loc[sel])
-    assert (chk[var] < 1e-7).all()
-
-    var = FV.P
-    sel = chk[var] >= 1e-5
-    print(f"\nCHECKING {var}\n", delta.loc[sel])
-    assert (chk[var] < 1e-5).all()
+    with foxes.Engine.new("threads", chunk_size_states=c):
+
+        algo = foxes.algorithms.Downwind(
+            farm,
+            states,
+            mbook=mbook,
+            rotor_model="centre",
+            wake_models=["Jensen_linear_k007"],
+            wake_frame="rotor_wd",
+            partial_wakes={"Jensen_linear_k007": "top_hat"},
+            verbosity=0,
+        )
+
+        data = algo.calc_farm()
+
+        df = data.to_dataframe()[[FV.WD, FV.AMB_REWS, FV.REWS, FV.AMB_P, FV.P]]
+
+        print("\nReading file", cfile)
+        fdata = pd.read_csv(cfile).set_index(["state", "turbine"])
+
+        print()
+        print("TRESULTS\n")
+        sel = (df[FV.P] > 0) & (fdata[FV.P] > 0)
+        df = df.loc[sel]
+        fdata = fdata.loc[sel]
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+
+        print("\nVERIFYING\n")
+        df[FV.WS] = df["REWS"]
+        df[FV.AMB_WS] = df["AMB_REWS"]
+
+        delta = df - fdata
+        print(delta)
+
+        chk = delta.abs()
+        print(chk.max())
+
+        var = FV.WS
+        print(f"\nCHECKING {var}")
+        sel = chk[var] >= 1e-7
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+        print(chk.loc[sel])
+        assert (chk[var] < 1e-7).all()
+
+        var = FV.P
+        sel = chk[var] >= 1e-5
+        print(f"\nCHECKING {var}\n", delta.loc[sel])
+        assert (chk[var] < 1e-5).all()
 
 
 if __name__ == "__main__":

tests/1_verification/flappy_0_6/row_Jensen_linear_tophat_IECTI2005/test_row_Jensen_linear_tophat_IECTI_2005.py

@@ -5,7 +5,7 @@ import inspect
 
 import foxes
 import foxes.variables as FV
-import foxes.constants as FC
+from foxes.config import config
 
 thisdir = Path(inspect.getfile(inspect.currentframe())).parent
 
@@ -22,8 +22,6 @@ def test():
     cfile = thisdir / "flappy" / "results.csv.gz"
     tfile = thisdir / "NREL-5MW-D126-H90.csv"
 
-    ck = {FC.STATE: c}
-
     mbook = foxes.models.ModelBook()
     ttype = foxes.models.turbine_types.PCtFile(
         data_source=tfile, var_ws_ct=FV.REWS, var_ws_P=FV.REWS
@@ -44,78 +42,79 @@ def test():
         verbosity=1,
     )
 
-    algo = foxes.algorithms.Downwind(
-        farm,
-        states,
-        rotor_model="centre",
-        mbook=mbook,
-        wake_models=["Jensen_linear_k007", "IECTI2005_max"],
-        wake_frame="rotor_wd",
-        partial_wakes=["top_hat", "top_hat"],
-        chunks=ck,
-        verbosity=1,
-    )
-
-    data = algo.calc_farm()
-
-    df = data.to_dataframe()[
-        [
-            FV.X,
-            FV.Y,
-            FV.WD,
-            FV.AMB_REWS,
-            FV.REWS,
-            FV.AMB_TI,
-            FV.TI,
-            FV.AMB_CT,
-            FV.CT,
+    with foxes.Engine.new("threads", chunk_size_states=c):
+
+        algo = foxes.algorithms.Downwind(
+            farm,
+            states,
+            rotor_model=rotor,
+            mbook=mbook,
+            wake_models=["Jensen_linear_k007", "IECTI2005_max"],
+            wake_frame="rotor_wd",
+            partial_wakes=["top_hat", "top_hat"],
+            verbosity=1,
+        )
+
+        data = algo.calc_farm()
+
+        df = data.to_dataframe()[
+            [
+                FV.X,
+                FV.Y,
+                FV.WD,
+                FV.AMB_REWS,
+                FV.REWS,
+                FV.AMB_TI,
+                FV.TI,
+                FV.AMB_CT,
+                FV.CT,
+            ]
         ]
-    ]
-
-    print("\nReading file", cfile)
-    fdata = pd.read_csv(cfile).set_index(["state", "turbine"])
-
-    print()
-    print("TRESULTS\n")
-    """sel = (df[FV.P] > 0) & (fdata[FV.P] > 0)
-    df = df.loc[sel]
-    fdata = fdata.loc[sel]"""
-    print(df)
-    print(fdata)
-
-    print("\nVERIFYING\n")
-    df[FV.WS] = df["REWS"]
-    df[FV.AMB_WS] = df["AMB_REWS"]
-
-    delta = df - fdata
-    print(delta)
-
-    chk = delta.abs()
-    print(chk.max())
-
-    var = FV.WS
-    print(f"\nCHECKING {var}")
-    sel = chk[var] >= 3e-3
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-    print(chk.loc[sel])
-    assert (chk[var] < 3e-3).all()
-
-    var = FV.TI
-    print(f"\nCHECKING {var}")
-    sel = chk[var] >= 3e-4
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-    print(chk.loc[sel])
-    assert (chk[var] < 3e-4).all()
-
-    var = FV.CT
-    print(f"\nCHECKING {var}")
-    sel = chk[var] >= 3e-5
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-    print(chk.loc[sel])
-    assert (chk[var] < 3e-5).all()
+
+        print("\nReading file", cfile)
+        fdata = pd.read_csv(cfile).set_index(["state", "turbine"])
+
+        print()
+        print("TRESULTS\n")
+        """sel = (df[FV.P] > 0) & (fdata[FV.P] > 0)
+        df = df.loc[sel]
+        fdata = fdata.loc[sel]"""
+        print(df)
+        print(fdata)
+
+        print("\nVERIFYING\n")
+        df[FV.WS] = df["REWS"]
+        df[FV.AMB_WS] = df["AMB_REWS"]
+
+        delta = df - fdata
+        print(delta)
+
+        chk = delta.abs()
+        print(chk.max())
+
+        var = FV.WS
+        print(f"\nCHECKING {var}")
+        sel = chk[var] >= 3e-3
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+        print(chk.loc[sel])
+        assert (chk[var] < 3e-3).all()
+
+        var = FV.TI
+        print(f"\nCHECKING {var}")
+        sel = chk[var] >= 3e-4
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+        print(chk.loc[sel])
+        assert (chk[var] < 3e-4).all()
+
+        var = FV.CT
+        print(f"\nCHECKING {var}")
+        sel = chk[var] >= 3e-5
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+        print(chk.loc[sel])
+        assert (chk[var] < 3e-5).all()
 
 
 if __name__ == "__main__":

tests/1_verification/flappy_0_6/row_Jensen_linear_tophat_IECTI2019/test_row_Jensen_linear_tophat_IECTI_2019.py

@@ -5,7 +5,7 @@ import inspect
 
 import foxes
 import foxes.variables as FV
-import foxes.constants as FC
+from foxes.config import config
 
 thisdir = Path(inspect.getfile(inspect.currentframe())).parent
 
@@ -22,8 +22,6 @@ def test():
     cfile = thisdir / "flappy" / "results.csv.gz"
     tfile = thisdir / "NREL-5MW-D126-H90.csv"
 
-    ck = {FC.STATE: c}
-
     mbook = foxes.models.ModelBook()
     ttype = foxes.models.turbine_types.PCtFile(
         data_source=tfile, var_ws_ct=FV.REWS, var_ws_P=FV.REWS
@@ -44,78 +42,79 @@ def test():
         verbosity=1,
     )
 
-    algo = foxes.algorithms.Downwind(
-        farm,
-        states,
-        mbook=mbook,
-        rotor_model=rotor,
-        wake_models=["Jensen_linear_k007", "IECTI2019_max"],
-        wake_frame="rotor_wd",
-        partial_wakes=["top_hat", "top_hat"],
-        chunks=ck,
-        verbosity=1,
-    )
-
-    data = algo.calc_farm()
-
-    df = data.to_dataframe()[
-        [
-            FV.X,
-            FV.Y,
-            FV.WD,
-            FV.AMB_REWS,
-            FV.REWS,
-            FV.AMB_TI,
-            FV.TI,
-            FV.AMB_CT,
-            FV.CT,
+    with foxes.Engine.new("threads", chunk_size_states=c):
+
+        algo = foxes.algorithms.Downwind(
+            farm,
+            states,
+            mbook=mbook,
+            rotor_model=rotor,
+            wake_models=["Jensen_linear_k007", "IECTI2019_max"],
+            wake_frame="rotor_wd",
+            partial_wakes=["top_hat", "top_hat"],
+            verbosity=1,
+        )
+
+        data = algo.calc_farm()
+
+        df = data.to_dataframe()[
+            [
+                FV.X,
+                FV.Y,
+                FV.WD,
+                FV.AMB_REWS,
+                FV.REWS,
+                FV.AMB_TI,
+                FV.TI,
+                FV.AMB_CT,
+                FV.CT,
+            ]
         ]
-    ]
-
-    print("\nReading file", cfile)
-    fdata = pd.read_csv(cfile).set_index(["state", "turbine"])
-
-    print()
-    print("TRESULTS\n")
-    """sel = (df[FV.P] > 0) & (fdata[FV.P] > 0)
-    df = df.loc[sel]
-    fdata = fdata.loc[sel]"""
-    print(df)
-    print(fdata)
-
-    print("\nVERIFYING\n")
-    df[FV.WS] = df["REWS"]
-    df[FV.AMB_WS] = df["AMB_REWS"]
-
-    delta = df - fdata
-    print(delta)
-
-    chk = delta.abs()
-    print(chk.max())
-
-    var = FV.WS
-    print(f"\nCHECKING {var}")
-    sel = chk[var] >= 3e-3
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-    print(chk.loc[sel])
-    assert (chk[var] < 3e-3).all()
-
-    var = FV.TI
-    print(f"\nCHECKING {var}")
-    sel = chk[var] >= 3e-4
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-    print(chk.loc[sel])
-    assert (chk[var] < 3e-4).all()
-
-    var = FV.CT
-    print(f"\nCHECKING {var}")
-    sel = chk[var] >= 3e-5
-    print(df.loc[sel])
-    print(fdata.loc[sel])
-    print(chk.loc[sel])
-    assert (chk[var] < 3e-5).all()
+
+        print("\nReading file", cfile)
+        fdata = pd.read_csv(cfile).set_index(["state", "turbine"])
+
+        print()
+        print("TRESULTS\n")
+        """sel = (df[FV.P] > 0) & (fdata[FV.P] > 0)
+        df = df.loc[sel]
+        fdata = fdata.loc[sel]"""
+        print(df)
+        print(fdata)
+
+        print("\nVERIFYING\n")
+        df[FV.WS] = df["REWS"]
+        df[FV.AMB_WS] = df["AMB_REWS"]
+
+        delta = df - fdata
+        print(delta)
+
+        chk = delta.abs()
+        print(chk.max())
+
+        var = FV.WS
+        print(f"\nCHECKING {var}")
+        sel = chk[var] >= 3e-3
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+        print(chk.loc[sel])
+        assert (chk[var] < 3e-3).all()
+
+        var = FV.TI
+        print(f"\nCHECKING {var}")
+        sel = chk[var] >= 3e-4
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+        print(chk.loc[sel])
+        assert (chk[var] < 3e-4).all()
+
+        var = FV.CT
+        print(f"\nCHECKING {var}")
+        sel = chk[var] >= 3e-5
+        print(df.loc[sel])
+        print(fdata.loc[sel])
+        print(chk.loc[sel])
+        assert (chk[var] < 3e-5).all()
 
 
 if __name__ == "__main__":