Use pytest 'raises'
Removed a few 'print'