def test_csv_has_header_sentiment():
"""Test function for recognising headers for small CSV file."""
with open(os.path.abspath('caterpillar/test_resources/twitter_sentiment.csv'), 'rbU') as f:
assert schema.csv_has_header(f.read(), csv.excel) is True
def test_csv_has_header_no_header():
"""Test function for recognising headers with a CSV file with no header row."""
with open(os.path.abspath('caterpillar/test_resources/test_no_header.csv'), 'rbU') as f:
assert schema.csv_has_header(f.read(), csv.excel) is False
