stroemung 0.1.2

A Computational Fluid Dynamics (CFD) simulator in Rust
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

#!/usr/bin/env python3

from pathlib import Path


import generate_test_data

import json


TEST_DATA_DIR = Path(Path(__file__).parent.resolve(), "test_data")


def load_file(filename):
    with open(filename, "r") as f:
        return json.load(f)


def load_expected_simulation(filename):
    data = load_file(filename)

    # Postprocess the flags field
    data["flags"] = [[generate_test_data.Flag(v) for v in x] for x in data["flags"]]

    return generate_test_data.SimulationOutput(**data)


def test_data_parsing():
    filename = "small_data.out"

    with open(Path(TEST_DATA_DIR, filename), "rb") as f:
        parsed_output = generate_test_data.parse_out_file(f)

    expected = load_expected_simulation(
        Path(TEST_DATA_DIR, filename + "_expected.json")
    )

    assert parsed_output == expected


def test_rust_grid_output():
    filename = "small_data.out"

    with open(Path(TEST_DATA_DIR, filename), "rb") as f:
        parsed_output = generate_test_data.parse_out_file(f).as_rust_grid()

    expected = load_file(Path(TEST_DATA_DIR, filename + "_rust_expected.json"))

    assert parsed_output == expected