def test_save_and_load(self): inputs = [] for x in range(0, 10): data = sorted([random.randint(0, 100) for i in range(0, 4)]) bar = Bar(timestamp=x, inst_id=3, open=data[1], high=data[3], low=data[0], close=data[2], vol=random.randint(100, 1000)) inputs.append(bar) self.db.save_bar(bar) self.db.stop() self.db = InMemoryDataStore() self.db.start(self.context) bars = self.db.load_all('bars') bars = sorted(bars, cmp=lambda x, y: x.timestamp - y.timestamp, reverse=False) self.assertEquals(10, len(bars)) for x in range(0, 10): self.assertEquals(inputs[x], bars[x])
def setUp(self): persistence_config = PersistenceConfig( None, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch) name = "test" create_at_start = True delete_at_stop = False app_config = ApplicationConfig( None, None, Clock.RealTime, persistence_config, InMemoryStoreConfig(file="%s_db.p" % name, create_at_start=create_at_start, delete_at_stop=delete_at_stop)) self.context = ApplicationContext(app_config=app_config) self.db = InMemoryDataStore() self.db.start(self.context)
def __init__(self): super(ProviderManager, self).__init__() self.add(Simulator()) self.add(IBBroker()) self.add(CassandraDataStore()) self.add(InfluxDataStore()) self.add(KDBDataStore()) self.add(MongoDBDataStore()) self.add(InMemoryDataStore()) self.add(CSVDataFeed()) self.add(PandasMemoryDataFeed()) self.add(GoogleDataFeed()) self.add(YahooDataFeed())
def setUp(self): persistence_config = PersistenceConfig(None, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch) name = "test" create_at_start = True delete_at_stop = False app_config = ApplicationConfig(None, None, Clock.RealTime, persistence_config, InMemoryStoreConfig(file="%s_db.p"%name, create_at_start=create_at_start, delete_at_stop=delete_at_stop)) self.context = ApplicationContext(app_config=app_config) self.db = InMemoryDataStore() self.db.start(self.context)
class InMemoryDBTest(TestCase): def setUp(self): persistence_config = PersistenceConfig( None, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch) name = "test" create_at_start = True delete_at_stop = False app_config = ApplicationConfig( None, None, Clock.RealTime, persistence_config, InMemoryStoreConfig(file="%s_db.p" % name, create_at_start=create_at_start, delete_at_stop=delete_at_stop)) self.context = ApplicationContext(app_config=app_config) self.db = InMemoryDataStore() self.db.start(self.context) def tearDown(self): self.db.remove_database() def test_save_and_load(self): inputs = [] for x in range(0, 10): data = sorted([random.randint(0, 100) for i in range(0, 4)]) bar = Bar(timestamp=x, inst_id=3, open=data[1], high=data[3], low=data[0], close=data[2], vol=random.randint(100, 1000)) inputs.append(bar) self.db.save_bar(bar) self.db.stop() self.db = InMemoryDataStore() self.db.start(self.context) bars = self.db.load_all('bars') bars = sorted(bars, cmp=lambda x, y: x.timestamp - y.timestamp, reverse=False) self.assertEquals(10, len(bars)) for x in range(0, 10): self.assertEquals(inputs[x], bars[x])
class InMemoryDBTest(TestCase): def setUp(self): persistence_config = PersistenceConfig(None, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch, DataStore.InMemoryDB, PersistenceMode.Batch) name = "test" create_at_start = True delete_at_stop = False app_config = ApplicationConfig(None, None, Clock.RealTime, persistence_config, InMemoryStoreConfig(file="%s_db.p"%name, create_at_start=create_at_start, delete_at_stop=delete_at_stop)) self.context = ApplicationContext(app_config=app_config) self.db = InMemoryDataStore() self.db.start(self.context) def tearDown(self): self.db.remove_database() def test_save_and_load(self): inputs = [] for x in range(0, 10): data = sorted([random.randint(0, 100) for i in range(0, 4)]) bar = Bar(timestamp=x, inst_id=3, open=data[1], high=data[3], low=data[0], close=data[2], vol=random.randint(100, 1000)) inputs.append(bar) self.db.save_bar(bar) self.db.stop() self.db = InMemoryDataStore() self.db.start(self.context) bars = self.db.load_all('bars') bars = sorted(bars, cmp=lambda x, y: x.timestamp - y.timestamp, reverse=False) self.assertEquals(10, len(bars)) for x in range(0, 10): self.assertEquals(inputs[x], bars[x])