1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
|
#! /usr/bin/env python
import openturns as ot
from openturns.testing import assert_almost_equal
from openturns.usecases import flood_model
ot.TESTPREAMBLE()
ot.PlatformInfo.SetNumericalPrecision(5)
print("Test the FloodModel data class.")
fm = flood_model.FloodModel()
# test parameter
assert fm.dim == 8
# test marginals means
assert_almost_equal(fm.Q.getMean()[0], 1338.1299666394023, 1e-12)
assert_almost_equal(fm.Ks.getMean()[0], 30.001003758482675, 1e-12)
assert_almost_equal(fm.Zv.getMean()[0], 50.0, 1e-12)
assert_almost_equal(fm.Zm.getMean()[0], 55.0, 1e-12)
assert_almost_equal(fm.B.getMean()[0], 300.0, 1e-12)
assert_almost_equal(fm.L.getMean()[0], 5000.0, 1e-12)
assert_almost_equal(fm.Zb.getMean()[0], 55.5, 1e-12)
assert_almost_equal(fm.Hd.getMean()[0], 3.0, 1e-12)
meanInput = fm.distribution.getMean()
print("meanInput = ", meanInput)
meanOutput = fm.model(meanInput)
print("meanOutput = ", meanOutput)
assert_almost_equal(meanOutput, [2.5313, -5.96869, 1.0362], 1.0e-4)
#
print("Test the simplified height model")
heightInputDistribution, heightModel = fm.getHeightModel()
meanInput = heightInputDistribution.getMean()
print("meanInput = ", meanInput)
meanOutput = heightModel(meanInput)
print("meanOutput = ", meanOutput)
assert_almost_equal(meanOutput, [2.5313])
print("Test the high Hd scenario")
fm = flood_model.FloodModel(distributionHdLow=False)
assert_almost_equal(fm.Hd.getMean()[0], 8.0, 1e-12)
|