40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
"""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)
|