Add tests on custom widgets