from numpy.testing import assert_allclose, assert_equal
from . import plt
from .. import utils


def test_path_data():
    circle = plt.Circle((0, 0), 1)
    vertices, codes = utils.SVG_path(circle.get_path())

    assert_allclose(vertices.shape, (25, 2))
    assert_equal(codes, ["M", "C", "C", "C", "C", "C", "C", "C", "C", "Z"])


def test_linestyle():
    linestyles = {
        "solid": "none",
        "-": "none",
        "dashed": "5.550000000000001,2.4000000000000004",
        "--": "5.550000000000001,2.4000000000000004",
        "dotted": "1.5,2.4749999999999996",
        ":": "1.5,2.4749999999999996",
        "dashdot": "9.600000000000001,2.4000000000000004,1.5,2.4000000000000004",
        "-.": "9.600000000000001,2.4000000000000004,1.5,2.4000000000000004",
        "": None,
        "None": None,
    }

    for ls, result in linestyles.items():
        (line,) = plt.plot([1, 2, 3], linestyle=ls)
        assert_equal(utils.get_dasharray(line), result)


def test_axis_w_fixed_formatter():
    positions, labels = [0, 1, 10], ["A", "B", "C"]

    plt.xticks(positions, labels)
    props = utils.get_axis_properties(plt.gca().xaxis)

    assert_equal(props["tickvalues"], positions)
    assert_equal(props["tickformat"], labels)