def test_portfolio_empty(self):
with patch.object(intrinio_data,
'get_latest_close_price',
return_value=('2019-08-31', 123.45)):
recommendation_set = SecurityRecommendationSet.from_dict(
self.sr_dict)
portfolio = Portfolio()
portfolio.create_empty_portfolio(recommendation_set)
self.assertTrue(portfolio.is_empty())
def test_portfolio_not_empty(self):
with patch.object(intrinio_data,
'get_latest_close_price',
return_value=('2019-08-31', 123.45)):
recommendation_set = SecurityRecommendationSet.from_dict(
self.sr_dict)
portfolio = Portfolio()
portfolio.create_empty_portfolio(recommendation_set)
portfolio.model['current_portfolio'] = {
'securities': [{
"ticker_symbol": "ABC",
"purchase_date": "2019-09-01T02:34:12.876012+00:00",
"purchase_price": 100,
"current_price": 200,
"trade_state": "FILLED",
"order_id": None
}]
}
self.assertFalse(portfolio.is_empty())
