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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
# ==================================================================================================================== #
# _____ _ _ _ _ _ _ _ _ #
# _ __ _ |_ _|__ ___ | (_)_ __ __ _ / \ | |_| |_ _ __(_) |__ _ _| |_ ___ ___ #
# | '_ \| | | || |/ _ \ / _ \| | | '_ \ / _` | / _ \| __| __| '__| | '_ \| | | | __/ _ \/ __| #
# | |_) | |_| || | (_) | (_) | | | | | | (_| |_ / ___ \ |_| |_| | | | |_) | |_| | || __/\__ \ #
# | .__/ \__, ||_|\___/ \___/|_|_|_| |_|\__, (_)_/ \_\__|\__|_| |_|_.__/ \__,_|\__\___||___/ #
# |_| |___/ |___/ #
# ==================================================================================================================== #
# Authors: #
# Patrick Lehmann #
# #
# License: #
# ==================================================================================================================== #
# Copyright 2017-2025 Patrick Lehmann - Bötzingen, Germany #
# Copyright 2007-2016 Patrick Lehmann - Dresden, Germany #
# #
# Licensed under the Apache License, Version 2.0 (the "License"); #
# you may not use this file except in compliance with the License. #
# You may obtain a copy of the License at #
# #
# http://www.apache.org/licenses/LICENSE-2.0 #
# #
# Unless required by applicable law or agreed to in writing, software #
# distributed under the License is distributed on an "AS IS" BASIS, #
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
# See the License for the specific language governing permissions and #
# limitations under the License. #
# #
# SPDX-License-Identifier: Apache-2.0 #
# ==================================================================================================================== #
#
"""
Helper functions for unittests.
"""
from contextlib import contextmanager
from io import StringIO
import sys as _sys
from typing import Generator, Dict, TypeVar, Tuple
K1 = TypeVar("K1")
V1 = TypeVar("V1")
K2 = TypeVar("K2")
V2 = TypeVar("V2")
def zip(dict1: Dict[K1, V1], dict2: Dict[K2, V2]) -> Generator[Tuple[K1, K2, V1, V2], None, None]:
l1 = len(dict1)
l2 = len(dict2)
if l1 != l2:
if l1 < l2:
raise ValueError(f"'dict1' (len={l1}) has less elements than 'dict2' (len={l2}).")
else:
raise ValueError(f"'dict1' (len={l1}) has more elements than 'dict2' (len={l2}).")
iter1 = iter(dict1.items())
iter2 = iter(dict2.items())
try:
while True:
key1, value1 = next(iter1)
key2, value2 = next(iter2)
yield key1, key2, value1, value2
except StopIteration:
return
@contextmanager
def CapturePrintContext():
old_out = _sys.stdout
old_err = _sys.stderr
try:
_sys.stdout = StringIO()
_sys.stderr = StringIO()
yield _sys.stdout, _sys.stderr
finally:
_sys.stdout = old_out
_sys.stderr = old_err
|
