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())