Adds support for custom tests.