diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/test_fix.py b/tests/test_fix.py new file mode 100644 index 0000000..45ac9e4 --- /dev/null +++ b/tests/test_fix.py @@ -0,0 +1,23 @@ +import csv_metadata_quality.fix as fix +import pytest + +def test_fix_leading_whitespace(): + '''Test fixing leading whitespace.''' + + value = ' Alan' + + assert fix.whitespace(value) == 'Alan' + +def test_fix_trailing_whitespace(): + '''Test fixing trailing whitespace.''' + + value = 'Alan ' + + assert fix.whitespace(value) == 'Alan' + +def test_fix_excessive_whitespace(): + '''Test fixing excessive whitespace.''' + + value = 'Alan Orth' + + assert fix.whitespace(value) == 'Alan Orth'