testing-python-exercise/tests/integration/test_diffusion2d.py at 657f1d0e8491224466ea7ebba63400922d030560 · Simulation-Software-Engineering/testing-python-exercise · 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
"""
Tests for functionality checks in class SolveDiffusion2D
"""

from diffusion2d import SolveDiffusion2D

import pytest
import numpy as np

def test_initialize_physical_parameters():
    """
    Checks function SolveDiffusion2D.initialize_domain
    """
    solver = SolveDiffusion2D()

    solver.initialize_domain()
    solver.initialize_physical_parameters(5.0, 300.0, 700.0)

    solver.initialize_physical_parameters(5.0, 300.0, 700.0)
    assert solver.dt == pytest.approx(0.0005, rel=1e-4)

    solver.initialize_physical_parameters(5.0, 500.0, 800.0)
    assert solver.dt == pytest.approx(0.0005, rel=1e-4)

    with pytest.raises(AssertionError):
        solver.initialize_physical_parameters(5, 500, 800)


def test_set_initial_condition():
    """
    Checks function SolveDiffusion2D.get_initial_function
    """
    solver = SolveDiffusion2D()

    solver.initialize_domain(8.0, 8.0, 1.0, 1.0)
    solver.initialize_physical_parameters(4.0, 0.0, 1.0)

    # 8x8 domain with circle at (5,5)
    expected_field = np.array(
        [[0., 0., 0., 0., 0., 0., 0., 0.],
         [0., 0., 0., 0., 0., 0., 0., 0.],
         [0., 0., 0., 0., 0., 0., 0., 0.],
         [0., 0., 0., 0., 0., 0., 0., 0.],
         [0., 0., 0., 0., 1., 1., 1., 0.],
         [0., 0., 0., 0., 1., 1., 1., 0.],
         [0., 0., 0., 0., 1., 1., 1., 0.],
         [0., 0., 0., 0., 0., 0., 0., 0.]]
    )

    actual_field = solver.set_initial_condition()

    error_norm = np.linalg.norm(expected_field - actual_field)
    assert 0 == pytest.approx(error_norm, abs=1e-14)