Test models

Ribasim developers use the following models in their testbench and in order to test new features.

Code

import ribasim_testmodels
import matplotlib.pyplot as plt

for model_name, model_constructor in ribasim_testmodels.constructors.items():
    if model_name.startswith("invalid"):
        continue

    model = model_constructor()
    fig, ax = plt.subplots(figsize = (6, 4))
    model.plot(ax)
    ax.set_title(label=model_name, loc="left")
    fig.text(0, 1, model_constructor.__doc__)
    ax.axis('off')
    plt.show()
    plt.close(fig)