Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
numerics_test.py 1.25 KB
Newer Older
mashun1's avatar
veros  
mashun1 committed
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 
from veros.core import numerics
from veros.pyom_compat import get_random_state

from test_base import compare_state


TEST_SETTINGS = dict(
    nx=70,
    ny=60,
    nz=50,
    dt_tracer=3600,
    dt_mom=3600,
    enable_cyclic_x=True,
    coord_degree=False,
    eq_of_state_type=1,
)


def test_calc_grid(pyom2_lib):
    vs_state, pyom_obj = get_random_state(pyom2_lib, extra_settings=TEST_SETTINGS)
    vs_state.variables.update(numerics.calc_grid(vs_state))
    pyom_obj.calc_grid()
    compare_state(vs_state, pyom_obj)


def test_calc_topo(pyom2_lib):
    vs_state, pyom_obj = get_random_state(pyom2_lib, extra_settings=TEST_SETTINGS)
    vs_state.variables.update(numerics.calc_topo(vs_state))
    pyom_obj.calc_topo()
    compare_state(vs_state, pyom_obj)


def test_calc_beta(pyom2_lib):
    vs_state, pyom_obj = get_random_state(pyom2_lib, extra_settings=TEST_SETTINGS)
    vs_state.variables.update(numerics.calc_beta(vs_state))
    pyom_obj.calc_beta()
    compare_state(vs_state, pyom_obj)


def test_calc_initial_conditions(pyom2_lib):
    vs_state, pyom_obj = get_random_state(pyom2_lib, extra_settings=TEST_SETTINGS)
    vs_state.variables.update(numerics.calc_initial_conditions(vs_state))
    pyom_obj.calc_initial_conditions()
    compare_state(vs_state, pyom_obj)