TST : improve again  test_kfold_balance test