Update test_max_shares.py