initial commit FESurrogateModelTutorial
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
import pandas as pd
|
||||
|
||||
from femsurrogate.data.bounds import DEFAULT_PARAMETER_BOUNDS
|
||||
from femsurrogate.data.sampling import generate_lhs_samples
|
||||
|
||||
|
||||
def test_lhs_samples_are_reproducible_for_fixed_seed():
|
||||
first = generate_lhs_samples(DEFAULT_PARAMETER_BOUNDS, n=8, seed=20260521)
|
||||
second = generate_lhs_samples(DEFAULT_PARAMETER_BOUNDS, n=8, seed=20260521)
|
||||
|
||||
pd.testing.assert_frame_equal(first, second)
|
||||
|
||||
|
||||
def test_lhs_samples_have_expected_columns_and_bounds():
|
||||
samples = generate_lhs_samples(DEFAULT_PARAMETER_BOUNDS, n=16, seed=20260521)
|
||||
|
||||
assert list(samples.columns) == list(DEFAULT_PARAMETER_BOUNDS)
|
||||
for column, bounds in DEFAULT_PARAMETER_BOUNDS.items():
|
||||
assert samples[column].between(bounds.lower, bounds.upper).all()
|
||||
Reference in New Issue
Block a user