Add tests for custom template dir