MAINT: reduce the number of samples, but the test is still too slow