Only include test code during testing

Avoid dependency on EUnit in production.