rscm 0.2.3

Rust Simple Climate Model - A framework for building reduced-complexity climate models
Documentation 
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

import numpy as np
import numpy.testing as npt

from rscm._lib import TwoLayerComponentBuilder
from rscm._lib.core import InterpolationStrategy, Model, Timeseries
from rscm.core import ModelBuilder


def test_model(time_axis):
    component = TwoLayerComponentBuilder.from_parameters(
        dict(
            lambda0=0.0,
            a=0.0,
            efficacy=0.0,
            eta=0.0,
            heat_capacity_deep=0.0,
            heat_capacity_surface=0.0,
        )
    ).build()

    builder = ModelBuilder()
    builder.with_time_axis(time_axis).with_rust_component(component)

    # Doesn't have any ERF data
    # Need pyo3_runtime.PanicException
    # with pytest.raises(Exception):
    #     builder.build()

    erf = Timeseries(
        np.asarray([1.0] * len(time_axis)),
        time_axis,
        "W / m^2",
        InterpolationStrategy.Next,
    )

    model = builder.with_exogenous_variable("Effective Radiative Forcing", erf).build()

    print(model.as_dot())

    model.run()


def test_model_serialisation(time_axis):
    component = TwoLayerComponentBuilder.from_parameters(
        dict(
            lambda0=0.0,
            a=0.0,
            efficacy=0.0,
            eta=0.0,
            heat_capacity_deep=0.0,
            heat_capacity_surface=0.0,
        )
    ).build()

    builder = ModelBuilder()
    builder.with_time_axis(time_axis).with_rust_component(component)
    erf = Timeseries(
        np.asarray([1.0] * len(time_axis)),
        time_axis,
        "W / m^2",
        InterpolationStrategy.Next,
    )

    model = builder.with_exogenous_variable("Effective Radiative Forcing", erf).build()

    model.step()

    serialised_model = model.to_toml()

    new_model = Model.from_toml(serialised_model)

    assert new_model.as_dot() == model.as_dot()
    assert new_model.current_time() == model.current_time()

    # Run the rest of the model and verify that the results are the same
    model.run()
    new_model.run()

    npt.assert_allclose(
        new_model.timeseries().get_timeseries_by_name("Surface Temperature").values(),
        model.timeseries().get_timeseries_by_name("Surface Temperature").values(),
    )