testing-python-exercise/tests/integration/test_diffusion2d.py at 2519a436cbd5237591b761eb410053d0b27a5202 · 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
"""
Tests for functionality checks in class SolveDiffusion2D
"""
import unittest
import numpy as np
from diffusion2d import SolveDiffusion2D


def test_initialize_physical_parameters():
    """
    Checks function SolveDiffusion2D.initialize_domain
    """
    solver = SolveDiffusion2D()
    w = 10.
    h = 10.
    dx = 1.
    dy = 1.
    d = 4.
    t_cold = 300.
    t_hot = 700.
    solver.initialize_domain(w, h, dx,dy)
    solver.initialize_physical_parameters(d, t_cold, t_hot)

    assert solver.dt == 0.0625

def test_set_initial_condition():
    """
    Checks function SolveDiffusion2D.get_initial_function
    """
    solver = SolveDiffusion2D()
    w = 4.
    h = 4.
    dx = 1.
    dy = 1.
    d = 4.
    t_cold = 300.
    t_hot = 700.
    solver.initialize_domain(w, h, dx, dy)
    solver.initialize_physical_parameters(d, t_cold, t_hot)

    u = solver.set_initial_condition()

    assert u.shape == (4, 4)
    assert u[0, 0] == 300.