import numpy as np
from helpers import assert_norm_equality, save

import dmsh


def generate(show=False):
    geo = dmsh.Rotation(dmsh.Rectangle(-1.0, +2.0, -1.0, +1.0), 0.1 * np.pi)
    X, cells = dmsh.generate(geo, 0.1, show=show, tol=1.0e-10, max_steps=100)
    return X, cells


def test(show=False):
    X, cells = generate(show)

    ref_norms = [9.4730152857365385e02, 3.1160562530932285e01, 2.2111300269652543e00]
    assert_norm_equality(X.flatten(), ref_norms, 1.0e-10)


if __name__ == "__main__":
    test(show=False)
    X, cells = generate(show=False)
    save("rotation.png", X, cells)