Python StockMarketData.get_number_of_companies示例

示例#1

文件： stock_market_data_test.py 项目： hijonga/oss-saki-ss19-exercise-4

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

示例#2

文件： stock_market_data_test.py 项目： hijonga/oss-saki-ss19-exercise-4

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