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