Update test_bagging.py