from grum.plotdesc import PlotDescription
import pytest

@pytest.mark.parametrize(
    "plot_name, plot_title, plot_xlabel, plot_ylabel, xs,ys", [('plot_name', 
    'plot_title', 
    'plot_xlabel', 
    'plot_ylabel', 
    [1, 2], 
    [3, 4]), 
    ('plot_name', 
    None, 
    None, 
    None, 
    [], 
    [])]
)
def test_plot_derscription_init(plot_name, plot_title, plot_xlabel, plot_ylabel, xs, ys):

    if plot_title:
        pd = PlotDescription(plot_name, plot_title,
                         plot_xlabel, plot_ylabel, xs, ys)
    else: 
        pd = PlotDescription(plot_name)
    
    assert pd.name == plot_name
    assert pd.title == plot_title
    assert pd.xlabel == plot_xlabel
    assert pd.ylabel == plot_ylabel
    assert pd.xs == xs
    assert pd.ys == ys
   
def test_data():
    pd = PlotDescription('name', xs = [1,2], ys = [3,4])

    assert pd.data == ([1,2], [3,4])

def test_append():
    pd = PlotDescription('name', xs = [1,2], ys = [3,4])

    pd.append([3,5])    

    assert pd.xs == [1,2,3]
    assert pd.ys == [3,4,5]