def test_load_from_db_one(self):
end_time = self.calc_end_time()
t = end_time
writer = PriceBoardDB.get_tf_writer('/tmp/onedata.tfrecords')
board = PriceBoardDB.load_from_db(t)
board.save_tf_to_writer(writer)
writer.close()
def test_load_from_db_600(self):
end_time = self.calc_end_time()
t = end_time
db = LogDb('/tmp/bitlog.db')
db.connect()
db.create_cursor()
writer = PriceBoardDB.get_tf_writer('/tmp/600rec.tfrecords')
retry = 600
while retry:
board = PriceBoardDB.load_from_connected_db(t-retry, db)
if board:
board.save_tf_to_writer(writer)
break
retry -= 1
db.close()
writer.close()