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)