I'd prefer to put tests in actually unit tests. Would you mind adding a formal test or two?