mud/tests/test_cli.py at main · mathematicalmichael/mud · GitHub

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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
# -*- coding: utf-8 -*-
"""
Tests for mud examples

"""
import pickle
from pathlib import Path

import numpy as np
from click.testing import CliRunner

from mud.cli import cli


def test_comparison_example(test_dir):
    runner = CliRunner()
    result = runner.invoke(
        cli, ["examples", "--save-path", str(test_dir), "comparison"]
    )
    assert result.exit_code == 0


def test_contours_example(test_dir):
    runner = CliRunner()
    result = runner.invoke(cli, ["examples", "--save-path", str(test_dir), "contours"])
    assert result.exit_code == 0


def test_high_dim_linear(test_dir):
    runner = CliRunner()
    result = runner.invoke(
        cli, ["examples", "--save-path", str(test_dir), "high-dim-linear"]
    )
    assert result.exit_code == 0


def test_wme_covariance(test_dir):
    runner = CliRunner()
    result = runner.invoke(
        cli, ["examples", "--save-path", str(test_dir), "wme-covariance"]
    )
    assert result.exit_code == 0


def test_poisson_generate(test_dir):
    runner = CliRunner()
    result = runner.invoke(
        cli,
        [
            "examples",
            "--seed",
            "21",
            "--save-path",
            test_dir,
            "poisson-generate",
            "5",
            "2",
        ],
    )

    fname = result.stdout[:-1].split("\n")[-1]

    if "Unable to run fenics" not in result.stdout:
        with open(fname, "rb") as fp:
            data = pickle.load(fp)

        assert result.exit_code == 0
        assert np.abs(data["true_vals"][0] + 0.48799728) < 0.001
        assert np.abs(data["true_vals"][1] - 0.00183782) < 0.001


def test_poisson_solve(test_dir):
    runner = CliRunner()
    data = str(Path(__file__).parent / "data" / "poisson_data")
    result = runner.invoke(
        cli,
        [
            "examples",
            "--save-path",
            str(test_dir),
            "--seed",
            "21",
            "poisson-solve",
            data,
        ],
    )
    assert result.exit_code == 0
    assert str(result.stdout) == "[-2.76754243 -1.6656349 ]\n"


def test_poisson_trials(test_dir):
    runner = CliRunner()
    data = str(Path(__file__).parent / "data" / "poisson_data")
    result = runner.invoke(
        cli,
        [
            "examples",
            "--save-path",
            str(test_dir),
            "--seed",
            "21",
            "poisson-trials",
            data,
            "-n",
            "2",
        ],
    )
    assert result.exit_code == 0
    assert "0.018693404000" in str(result.stdout)


def test_adcirc_solve(test_dir):
    runner = CliRunner()
    data = str(Path(__file__).parent / "data" / "adcirc_data")
    result = runner.invoke(
        cli,
        [
            "examples",
            "--save-path",
            str(test_dir),
            "--seed",
            "21",
            "adcirc-solve",
            data,
            "-p",
            "all",
            "-t1",
            "2018-01-01T10:03:00.000000000",
            "-t2",
            "2018-01-01T12:33:00.000000000",
        ],
    )
    assert result.exit_code == 0
    assert "[0.05266253 0.00294599]" in str(result.stdout)


def test_mud_paper(test_dir):
    runner = CliRunner()
    result = runner.invoke(cli, ["examples", "--save-path", str(test_dir), "mud-paper"])
    assert result.exit_code == 0