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