44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
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] |