blob: c718fa379a96b452d39ffbd0184d7681f86ae59d (
plain)
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
|
"""Utility used for unit test."""
from typing import Dict, Sequence
class MockNode:
"""Mock node class."""
def __init__(self, attrs: Dict[str, str],
child_attrs: Sequence[Dict[str, str]] = ({},),
is_child: bool = False):
self.attributes = attrs
self.childs = [
MockNode(attrs, is_child=True) for attrs in child_attrs if attrs]
self.is_child = is_child
if 'bounds' not in self.attributes:
self.attributes['bounds'] = '[0,0][384,384]'
def __str__(self):
xml_str_elements = []
if not self.is_child:
xml_str_elements.append(
"<?xml version='1.0' encoding='UTF-8' standalone='yes' ?>")
xml_str_elements.append('<hierarchy rotation="0">')
xml_str_elements.append('<node index="0"')
for attr_key, attr_val in self.attributes.items():
xml_str_elements.append(f' {attr_key}="{attr_val}"')
xml_str_elements.append('>')
for child in self.childs:
xml_str_elements.append(str(child))
xml_str_elements.append('</node>')
if not self.is_child:
xml_str_elements.append('</hierarchy>')
return ''.join(xml_str_elements)
|
