import pytest
from helpers import assert_norm_equality, save

import dmsh

def generate_ellipse(show=False):
    geo = dmsh.Ellipse([0.0, 0.0], 2.0, 1.0)
    X, cells = dmsh.generate(geo, 0.2, show=show)

    geo.plot()

    return X, cells

@pytest.mark.skip
def test_ellipse(show=False):
    X, cells = generate_ellipse(show)
    ref_norms = [2.5108941453435716e02, 1.5652963447587933e01, 1.9890264390440919e00]
    assert_norm_equality(X.flatten(), ref_norms, 2.0e-2)


if __name__ == "__main__":
    X, cells = generate_ellipse(show=True)
    save("ellipse.png", X, cells)