1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
import logging
from pprint import pformat
import pytest
from pbcommand.models.report import PlotGroup, Plot, PbReportError
log = logging.getLogger(__name__)
class TestPlotGroup:
def test_init(self):
"""Test constructor with kwargs"""
plot = Plot('a_plot', 'path/to/image.png', caption="My Image")
p = PlotGroup('my_pg', plots=[plot])
assert p is not None
def test_plotgroup_null_id(self):
"""Can't create an plotGroup without an id."""
def _test():
p = PlotGroup(None)
with pytest.raises(PbReportError):
_test()
def test_plotgroup_add_duplicate_plot(self):
"""Can't add plots with duplicate ids."""
def _test():
pg = PlotGroup('foo')
pg.add_plot(Plot('id', 'i1'))
pg.add_plot(Plot('id', 'i2'))
with pytest.raises(PbReportError):
_test()
def test_to_dict(self):
"""Test plotGroup to_dict function."""
a = PlotGroup('123', title='foo title', legend='foo legend',
thumbnail='foo thumbnail')
a.add_plot(Plot('id', 'i1', caption='a caption'))
d = a.to_dict()
log.debug(pformat(d))
assert '123' == d['id']
assert 'foo title' == d['title']
assert 'foo legend' == d['legend']
assert 'foo thumbnail' == d['thumbnail']
assert 1 == len(d['plots'])
log.info(a)
assert repr(a) is not None
def test_adding_incorrect_type(self):
"""Validate type when adding Plots."""
def _test():
plots = ['Not a plot instance', 'Another bad plot.']
p = PlotGroup('my_plotgroup', plots=plots)
with pytest.raises(TypeError):
_test()
|
