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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
import pytest
from petl.test.helpers import ieq, eq_
import petl.config as config
def assert_failonerror(input_fn, expected_output):
"""In the input rows, the first row should process through the
transformation cleanly. The second row should generate an
exception. There are no requirements for any other rows."""
#=========================================================
# Test function parameters with default config settings
#=========================================================
# test the default config setting: failonerror == False
eq_(config.failonerror, False)
# By default, a bad conversion does not raise an exception, and
# values for the failed conversion are returned as None
table2 = input_fn()
ieq(expected_output, table2)
ieq(expected_output, table2)
# When called with failonerror is False or None, a bad conversion
# does not raise an exception, and values for the failed conversion
# are returned as None
table3 = input_fn(failonerror=False)
ieq(expected_output, table3)
ieq(expected_output, table3)
table3 = input_fn(failonerror=None)
ieq(expected_output, table3)
ieq(expected_output, table3)
# When called with failonerror=True, a bad conversion raises an
# exception
with pytest.raises(Exception):
table4 = input_fn(failonerror=True)
table4.nrows()
# When called with failonerror='inline', a bad conversion
# does not raise an exception, and an Exception for the failed
# conversion is returned in the result.
expect5 = expected_output[0], expected_output[1]
table5 = input_fn(failonerror='inline')
ieq(expect5, table5.head(1))
ieq(expect5, table5.head(1))
excp = table5[2][0]
assert isinstance(excp, Exception)
#=========================================================
# Test config settings
#=========================================================
# Save config setting
saved_config_failonerror = config.failonerror
# When config.failonerror == True, a bad conversion raises an
# exception
config.failonerror = True
with pytest.raises(Exception):
table6 = input_fn()
table6.nrows()
# When config.failonerror == 'inline', a bad conversion
# does not raise an exception, and an Exception for the failed
# conversion is returned in the result.
expect7 = expected_output[0], expected_output[1]
config.failonerror = 'inline'
table7 = input_fn()
ieq(expect7, table7.head(1))
ieq(expect7, table7.head(1))
excp = table7[2][0]
assert isinstance(excp, Exception)
# When config.failonerror is an invalid value, but still truthy, it
# behaves the same as if == True
config.failonerror = 'invalid'
with pytest.raises(Exception):
table8 = input_fn()
table8.nrows()
# When config.failonerror is None, it behaves the same as if
# config.failonerror is False
config.failonerror = None
table9 = input_fn()
ieq(expected_output, table9)
ieq(expected_output, table9)
# A False keyword parameter overrides config.failonerror == True
config.failonerror = True
table10 = input_fn(failonerror=False)
ieq(expected_output, table10)
ieq(expected_output, table10)
# A None keyword parameter uses config.failonerror == True
config.failonerror = True
with pytest.raises(Exception):
table11 = input_fn(failonerror=None)
table11.nrows()
# restore config setting
config.failonerror = saved_config_failonerror
|
