Add some tests for the auto_balance attribute