def test_profit(self): wallet = Wallet(initial_balance) wallet.push(BUY_tr) wallet.push(SELL_tr) self.assertEqual(wallet.balance, initial_balance + 1) wallet.reset() wallet.push(BUY_tr) wallet.push(BUY_tr) wallet.push(SELL_tr) self.assertEqual(wallet.balance, initial_balance + 1)
def test_reset(self): wallet = Wallet(initial_balance) transaction1 = Transaction('SPCE', 'buy', 10, 10.14) wallet.push(transaction1) transaction2 = Transaction('TSLA', 'buy', 2, 100.14) wallet.push(transaction2) wallet.reset() self.assertEqual(initial_balance, wallet.balance) self.assertTrue(len(wallet._portfolio) == 0) self.assertTrue(len(wallet._ledger) == 0)