FilesExpand file tree

master

/

utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

46 lines (34 loc) · 1.17 KB

master

/

utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

46 lines (34 loc) · 1.17 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

import six

from lxml import etree

from six import binary_type, string_types

def load_xml(xml):

parser = etree.XMLParser(

remove_blank_text=True, remove_comments=True, resolve_entities=False

)

return etree.fromstring(xml.strip(), parser=parser)

def assert_nodes_equal(result, expected):

def _convert_node(node):

if isinstance(node, (string_types, binary_type)):

return load_xml(node)

return node

# assert node_1 == node_2

result = etree.tostring(_convert_node(result), pretty_print=True)

expected = etree.tostring(_convert_node(expected), pretty_print=True)

if six.PY3:

result = result.decode("utf-8")

expected = expected.decode("utf-8")

assert result == expected

def render_node(element, value):

node = etree.Element("document")

element.render(node, value)

return node

class DummyTransport(object):

def __init__(self):

self._items = {}

def bind(self, url, node):

self._items[url] = node

def load(self, url):

data = self._items[url]

if isinstance(data, (binary_type, string_types)):

return data

return etree.tostring(data)