Remove print() statements from unit tests

These have been moved to failure messages, as per review feedback.