Updated the test interface.

Added an argument to allow for code specific configuration.