My Transaction class has a method \'balance\' which calls a method \'balance_after\' from my Account class. I would like to make a test where I can check the \'balance_after