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
|
from __future__ import absolute_import, print_function, division
import os
import sys
import pytest
from petl.compat import izip_longest
def eq_(expect, actual, msg=None):
"""Test when two values from a python variable are exactly equals (==)"""
assert expect == actual, msg or ('%r != %s' % (expect, actual))
def assert_almost_equal(first, second, places=None, msg=None):
"""Test when the values are aproximatedly equals by a places exponent"""
vabs = None if places is None else 10 ** (- places)
assert pytest.approx(first, second, abs=vabs), msg
def ieq(expect, actual, cast=None):
"""Test when values of a iterable are equals for each row and column"""
ie = iter(expect)
ia = iter(actual)
ir = 0
for re, ra in izip_longest(ie, ia, fillvalue=None):
if cast:
ra = cast(ra)
if re is None and ra is None:
continue
if type(re) in (int, float, bool, str):
eq_(re, ra)
continue
_ieq_row(re, ra, ir)
ir = ir + 1
def _ieq_row(re, ra, ir):
assert ra is not None, "Expected row #%d is None, but result row is not None" % ir
assert re is not None, "Expected row #%d is not None, but result row is None" % ir
ic = 0
for ve, va in izip_longest(re, ra, fillvalue=None):
if isinstance(ve, list):
for je, ja in izip_longest(ve, va, fillvalue=None):
_ieq_col(je, ja, re, ra, ir, ic)
elif not isinstance(ve, dict):
_ieq_col(ve, va, re, ra, ir, ic)
ic = ic + 1
def _ieq_col(ve, va, re, ra, ir, ic):
"""Print two values when they aren't exactly equals (==)"""
try:
eq_(ve, va)
except AssertionError as ea:
# Show the values but only when they differ
print('\nrow #%d' % ir, re, ' != ', ra, file=sys.stderr)
print('col #%d: ' % ic, ve, ' != ', va, file=sys.stderr)
raise ea
def ieq2(expect, actual, cast=None):
"""Test when iterables values are equals twice looking for side effects"""
ieq(expect, actual, cast)
ieq(expect, actual, cast)
def get_env_vars_named(prefix, remove_prefix=True):
"""Get all named variables starting with prefix"""
res = {}
varlen = len(prefix)
for varname, varvalue in os.environ.items():
if varname.upper().startswith(prefix.upper()):
if remove_prefix:
varname = varname[varlen:]
res[varname] = varvalue
if len(res) == 0:
return None
return res
|
