Maybe you can only test `max_features` in `[2, 4]` to speed up the test.