updated tests for a module instead of a class.