Source code for openpyxl.chart.tests.test_data_source

from __future__ import absolute_import
# Copyright (c) 2010-2017 openpyxl

import pytest

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


@pytest.fixture
[docs]def NumRef(): from ..data_source import NumRef return NumRef
[docs]class TestNumRef:
[docs] def test_from_xml(self, NumRef): src = """ <numRef> <f>Blatt1!$A$1:$A$12</f> </numRef> """ node = fromstring(src) num = NumRef.from_tree(node) assert num.ref == "Blatt1!$A$1:$A$12"
[docs] def test_to_xml(self, NumRef): num = NumRef(f="Blatt1!$A$1:$A$12") xml = tostring(num.to_tree("numRef")) expected = """ <numRef> <f>Blatt1!$A$1:$A$12</f> </numRef> """ diff = compare_xml(xml, expected) assert diff is None, diff
@pytest.fixture
[docs]def StrRef(): from ..data_source import StrRef return StrRef
[docs]class TestStrRef:
[docs] def test_ctor(self, StrRef): data_source = StrRef(f="Sheet1!A1") xml = tostring(data_source.to_tree()) expected = """ <strRef> <f>Sheet1!A1</f> </strRef> """ diff = compare_xml(xml, expected) assert diff is None, diff
[docs] def test_from_xml(self, StrRef): src = """ <strRef> <f>'Render Start'!$A$2</f> </strRef> """ node = fromstring(src) data_source = StrRef.from_tree(node) assert data_source == StrRef(f="'Render Start'!$A$2")
@pytest.fixture
[docs]def StrVal(): from ..data_source import StrVal return StrVal
[docs]class TestStrVal:
[docs] def test_ctor(self, StrVal): val = StrVal(v="something") xml = tostring(val.to_tree()) expected = """ <strVal idx="0"> <v>something</v> </strVal> """ diff = compare_xml(xml, expected) assert diff is None, diff
[docs] def test_from_xml(self, StrVal): src = """ <pt idx="4"> <v>else</v> </pt> """ node = fromstring(src) val = StrVal.from_tree(node) assert val == StrVal(idx=4, v="else")
@pytest.fixture
[docs]def StrData(): from ..data_source import StrData return StrData
[docs]class TestStrData:
[docs] def test_ctor(self, StrData): data_source = StrData(ptCount=1) xml = tostring(data_source.to_tree()) expected = """ <strData> <ptCount val="1"></ptCount> </strData> """ diff = compare_xml(xml, expected) assert diff is None, diff
[docs] def test_from_xml(self, StrData): src = """ <strData> <ptCount val="4"></ptCount> </strData> """ node = fromstring(src) data_source = StrData.from_tree(node) assert data_source == StrData(ptCount=4)