Add (optional) unit test infrastructure