refactor to use a function call for tests, and to have arbitrary args