def setUp(self): Base.metadata.create_all() instrument = Instrument() instrument.ticker = 'MyTicker' instrument.asset_class = 'MyAssetClass' instrument.name = 'MyName' instrument.currency = 'USD' instrument.type = 'MyType' instrument.transactions_fees = 0. instrument.point_value = 1 instrument.exchange = 'MyExchange' instrument.rolling_month = 'H' market_data = MarketData() market_data.instrument = instrument market_data.type = DataType.CLOSE market_data.date = datetime(2000,1,3) market_data.value = 100. session = Session() instrument_dao = InstrumentDAO() instrument_dao.save(session, instrument) market_data_dao = MarketDataDAO() market_data_dao.save(session, market_data) self.instrument_service = InstrumentService()
def testDAO(self): data = MarketData() data.type = self.type data.value = self.value data.date = self.date data.instrument = self.instrument market_data_dao = MarketDataDAO() # test save market_data_dao.save(self.session, data) self.assertNotEqual(data.id, None, 'market data id') self.assertEqual(data.type, self.type, 'market data type') self.assertEqual(data.value, self.value, 'market data value') self.assertEqual(data.date, self.date, 'market data date') self.assertEqual(data.instrument_id, self.instrument.id, 'market data type') # test get by instrument new_data = market_data_dao.get_by_instrument(self.session, self.instrument) self.assertEqual(new_data[0].id, data.id, 'both ids equals') last_date = market_data_dao.get_last_date(self.session, self.instrument) self.assertEqual(last_date, data.date) # test delete market_data_dao.delete(self.session, data)