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
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.")
