def testAddAndRemoveFunds(self): myAccount = Account("Lukas' Savings", 1000, "Savings") expected = 1000 myAccount.removeFunds(100) myAccount.addFunds(100) actual = myAccount.getAmount() self.assertEqual(expected, actual)