def test_stock_market_data_one_company_one_period(self): stock_market_data = StockMarketData([Company.A], [Period.TRAINING]) self.assertIsNotNone(stock_market_data) self.assertEqual(stock_market_data.get_number_of_companies(), 1) self.assertEqual(stock_market_data.get_row_count(), 12588) self.assertEqual(stock_market_data.get_most_recent_trade_day(), Date(2011, 12, 30)) self.assertEqual(stock_market_data.get_most_recent_price(Company.A), 34.802376) self.assertIsNone(stock_market_data.get_most_recent_price(Company.B))
def test_stock_market_data_one_company_two_periods(self): stock_market_data = StockMarketData([Company.A], [Period.TRAINING, Period.TESTING]) self.assertIsNotNone(stock_market_data) self.assertEqual(stock_market_data.get_number_of_companies(), 1) self.assertEqual(stock_market_data.get_row_count(), 13594) self.assertEqual(stock_market_data.get_most_recent_trade_day(), Date(2015, 12, 31)) self.assertEqual(stock_market_data.get_most_recent_price(Company.A), 102.759895) self.assertIsNone(stock_market_data.get_most_recent_price(Company.B))
def test_get_row_count(self): stock_market_data = StockMarketData([Company.A, Company.B], [Period.TRAINING, Period.TESTING]) self.assertEqual(stock_market_data.get_row_count(), stock_market_data[Company.A].get_row_count())