Add tests for supervised tasks