def setup_files(dir_name, gps_start_time, gps_end_time): # Filter out the ones that are outside our time range xml_files = segmentdb_utils.get_all_files_in_range(dir_name, gps_start_time, gps_end_time) handle, temp_db = tempfile.mkstemp(suffix='.sqlite') os.close(handle) target = dbtables.get_connection_filename(temp_db, None, True, False) connection = ligolw_sqlite.setup(target) ligolw_sqlite.insert_from_urls(connection, xml_files) # [temp_xml]) segmentdb_utils.ensure_segment_table(connection) return temp_db, connection
[('H1', 'DMT-TESTSEG_2', 1, 924900000, 924900016, 0, 0), ('H1', 'DMT-TESTSEG_3', 1, 924900000, 924900016, 0, 0)]) if res[0] != segmentlist([segment(924900000, 924900010)]): return False if res[1] != segmentlist([segment(924900008, 924900016)]): return False return True if __name__ == '__main__': db_name = 'seg_test_db.sqlite' target = dbtables.get_connection_filename(db_name, None, True, False) connection = ligolw_sqlite.setup(target) engine = query_engine.SqliteQueryEngine(connection) ligolw_sqlite.insert(connection, ['test_segdb_utils.xml']) print "Testing basic segment summary...", print test_basic_seg_summary(engine) and "suceeded." or "FAILED." print "Testing expanding version numbers...", print test_expand_versions(engine) and "suceeded." or "FAILED." print "Testing optimized segment query...", print test_optimized_query(engine) and "suceeded." or "FAILED." connection.close()
res = segmentdb_utils.query_segments( engine, 'segment_summary', [ ('H1','DMT-TESTSEG_2',1,924900000,924900016,0,0), ('H1','DMT-TESTSEG_3',1,924900000,924900016,0,0) ] ) if res[0] != segmentlist([segment(924900000, 924900010)]): return False if res[1] != segmentlist([segment(924900008, 924900016)]): return False return True if __name__ == '__main__': db_name = 'seg_test_db.sqlite' target = dbtables.get_connection_filename(db_name, None, True, False) connection = ligolw_sqlite.setup(target) engine = query_engine.SqliteQueryEngine(connection) ligolw_sqlite.insert(connection, ['test_segdb_utils.xml']) print("Testing basic segment summary...", end=' ') print(test_basic_seg_summary(engine) and "suceeded." or "FAILED.") print("Testing expanding version numbers...", end=' ') print(test_expand_versions(engine) and "suceeded." or "FAILED.") print("Testing optimized segment query...", end=' ') print(test_optimized_query(engine) and "suceeded." or "FAILED.")