Source code for openpyxl.chart.tests.test_legend

from __future__ import absolute_import

# Copyright (c) 2010-2017 openpyxl
import pytest

from openpyxl.xml.functions import fromstring, tostring
from openpyxl.tests.helper import compare_xml

@pytest.fixture
[docs]def Legend():
    from ..legend import Legend
    return Legend


[docs]class TestLegend:

[docs]    def test_ctor(self, Legend):
        legend = Legend()
        xml = tostring(legend.to_tree())
        expected = """
        <legend>
          <legendPos val="r" />
        </legend>
        """
        diff = compare_xml(xml, expected)
        assert diff is None, diff


[docs]    def test_from_xml(self, Legend):
        src = """
        <legend>
          <legendPos val="r" />
        </legend>
        """
        node = fromstring(src)
        legend = Legend.from_tree(node)
        assert legend == Legend()


@pytest.fixture
[docs]def LegendEntry():
    from ..legend import LegendEntry
    return LegendEntry


[docs]class TestLegendEntry:

[docs]    def test_ctor(self, LegendEntry):
        legend = LegendEntry()
        xml = tostring(legend.to_tree())
        expected = """
        <legendEntry>
          <idx val="0"></idx>
        </legendEntry>
        """
        diff = compare_xml(xml, expected)
        assert diff is None, diff


[docs]    def test_from_xml(self, LegendEntry):
        src = """
        <legendEntry>
          <idx val="0"></idx>
        </legendEntry>
        """
        node = fromstring(src)
        legend = LegendEntry.from_tree(node)
        assert legend == LegendEntry()