def test_ticker_gains_ok(transactions, exchange_rates_mock):
tg = TickerGains(transactions[0].ticker)
er = ExchangeRate('USD', transactions[0].date, transactions[3].date)
er_map = {'USD': er}
# Add first transaction - 'BUY'
# Add second transaction - 'BUY'
# Add third transaction - 'SELL'
transactions_to_test = [transactions[0], transactions[2], transactions[3]]
tg.add_transactions(transactions_to_test, er_map)
assert transactions[0].exchange_rate == 2.0
assert transactions[0].share_balance == 100
assert transactions[0].proceeds == 10000.0
assert transactions[0].capital_gain == 0.0
assert transactions[0].acb == 10020.00
assert transactions[0].superficial_loss is False
assert transactions[0].expenses == 20.0
assert transactions[2].exchange_rate == 2.0
assert transactions[2].share_balance == 50
assert transactions[2].proceeds == 12000.00
assert transactions[2].capital_gain == 6970.00
assert transactions[2].acb == 5010.00
assert transactions[2].superficial_loss is False
assert transactions[2].expenses == 20.0
assert transactions[3].exchange_rate == 2.0
assert transactions[3].share_balance == 100
assert transactions[3].proceeds == 13000.00
assert transactions[3].capital_gain == 0.0
assert transactions[3].acb == 13020.00
assert transactions[3].superficial_loss is False
assert transactions[3].expenses == 20.0
def calculate_gains(transactions, year, ticker):
ticker_transactions = transactions.filter_by(tickers=[ticker],
max_year=year)
er_map = _get_map_of_currencies_to_exchange_rates(ticker_transactions)
tg = TickerGains(ticker)
tg.add_transactions(ticker_transactions, er_map)
return ticker_transactions.filter_by(year=year,
action='SELL',
superficial_loss=False)
def test_ticker_gains_negative_balance(transactions, exchange_rates_mock):
"""If the first transaction added is a sell, it is illegal since
this causes a negative balance, which is impossible"""
sell_transaction = transactions[2]
tg = TickerGains(sell_transaction.ticker)
er = ExchangeRate('USD', transactions[2].date, transactions[2].date)
er_map = {'USD': er}
with pytest.raises(ClickException) as excinfo:
tg.add_transactions([sell_transaction], er_map)
assert excinfo.value.message == "Transaction caused negative share balance"
def test_superficial_loss_no_purchase_after_loss(exchange_rates_mock):
"""Testing if transaction is marked as a superficial loss even if
there are no purchases made after the loss"""
transactions = [
Transaction(date(2018, 1, 1), 'ESPP PURCHASE', 'ANET', 'BUY', 100,
100.00, 10.00, 'USD'),
Transaction(date(2018, 1, 2), 'RSU VEST', 'ANET', 'SELL', 99, 50.00,
10.00, 'USD')
]
tg = TickerGains(transactions[0].ticker)
er = ExchangeRate('USD', transactions[0].date, transactions[1].date)
er_map = {'USD': er}
tg.add_transactions(transactions, er_map)
assert transactions[1].superficial_loss
assert transactions[1].capital_gain == 0
def test_gain_not_marked_as_superficial_loss(exchange_rates_mock):
"""Testing if transaction is not marked as a superficial loss if
it does not result in a loss"""
transactions = [
Transaction(date(2018, 1, 1), 'ESPP PURCHASE', 'ANET', 'BUY', 100,
1.00, 10.00, 'USD'),
Transaction(date(2018, 8, 1), 'RSU VEST', 'ANET', 'SELL', 100, 50.00,
10.00, 'USD')
]
tg = TickerGains(transactions[0].ticker)
er = ExchangeRate('USD', transactions[0].date, transactions[1].date)
er_map = {'USD': er}
tg.add_transactions(transactions, er_map)
assert not transactions[1].superficial_loss
assert transactions[1].capital_gain > 0
def test_loss_no_balance_after_window(exchange_rates_mock):
"""Testing if transaction is not marked as a superficial loss if
there is no share balance 30 days after the loss"""
transactions = [
Transaction(date(2018, 1, 1), 'ESPP PURCHASE', 'ANET', 'BUY', 100,
100.00, 10.00, 'USD'),
Transaction(date(2018, 1, 2), 'RSU VEST', 'ANET', 'SELL', 99, 50.00,
10.00, 'USD'),
Transaction(date(2018, 1, 3), 'RSU VEST', 'ANET', 'SELL', 1, 100.00,
10.00, 'USD'),
Transaction(date(2018, 2, 10), 'RSU VEST', 'ANET', 'BUY', 1, 100.00,
10.00, 'USD')
]
tg = TickerGains(transactions[0].ticker)
er = ExchangeRate('USD', transactions[0].date, transactions[1].date)
er_map = {'USD': er}
tg.add_transactions(transactions, er_map)
assert not transactions[1].superficial_loss
assert transactions[1].capital_gain < 0