def test_parse_headers6():
t = BaseTransform()
t.header_map = {'A': 'FIRST NAME', 'B': 'LAST NAME'}
t.headers = ['A', 'B', 'ADDRESS']
t.valid_headers = ['FIRST NAME', 'LAST NAME', 'SKIP']
t.default_not_found = 'SKIP'
assert ['FIRST NAME', 'LAST NAME', 'SKIP'] == t.parse_headers()
def test_parse_headers7():
t = BaseTransform()
t.header_map = {'A': 'FIRST NAME', 'Column B': 'LAST NAME', 'C': 'NOTES'}
t.headers = ['A', 'Column B', 'C']
t.valid_headers = ['FIRST NAME', 'LAST NAME', 'NOTES']
t.prefixable_headers.append('NOTES')
assert ['FIRST NAME', 'LAST NAME', 'NOTES'] == t.parse_headers()
assert t.column_mods[2].prefix == 'C: '
def test_parse_headers5():
t = BaseTransform()
t.header_map = {'A': 'FIRST NAME', 'B': 'LAST NAME'}
t.headers = ['A', 'B', 'ADDRESS']
t.valid_headers = ['FIRST NAME', 'LAST NAME', 'ADDRESS']
assert ['FIRST NAME', 'LAST NAME', 'ADDRESS'] == t.parse_headers()
def test_parse_headers4():
t = BaseTransform()
t.header_map = {'A': 'NAME', 'B': 'ADDRESS'}
t.headers = [None, 'B']
assert ['', 'ADDRESS'] == t.parse_headers()
def test_parse_headers3():
t = BaseTransform()
t.header_map = {'A': 'NAME', 'B': 'ADDRESS'}
assert t.parse_headers() is None
def test_parse_headers2():
t = BaseTransform()
t.headers = ['A', 'B']
assert ['A', 'B'] == t.parse_headers()