Add more unit tests for modules and lint

We need to continue adding tests for our modules and the lint stage. Until we've added tests for all functions/methods for lint and modules, we'll need to keep returning to this refactored code to complete tests for it.