examples/natmod/features2/test.py


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

# This Python code will be merged with the C code in main.c

# ruff: noqa: F821 - this file is evaluated with C-defined names in scope

import array


def isclose(a, b):
    return abs(a - b) < 1e-3


def test():
    tests = [
        isclose(add(0.1, 0.2), 0.3),
        isclose(add_f(0.1, 0.2), 0.3),
    ]

    ar = array.array("f", [1, 2, 3.5])
    productf(ar)
    tests.append(isclose(ar[0], 7))

    if "add_d" in globals():
        tests.append(isclose(add_d(0.1, 0.2), 0.3))

    print(tests)

    if not all(tests):
        raise SystemExit(1)


test()