Mercurial Hosting > traffic-intelligence
changeset 212:ce44605f888a
minor modifications
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 06 Jun 2012 23:39:03 -0400 |
parents | ada6e8fbe4c6 |
children | 5cde6da74605 |
files | python/storage.py |
diffstat | 1 files changed, 5 insertions(+), 8 deletions(-) [+] |
line wrap: on
line diff
--- a/python/storage.py Wed Jun 06 13:24:04 2012 -0400 +++ b/python/storage.py Wed Jun 06 23:39:03 2012 -0400 @@ -25,9 +25,9 @@ connection = sqlite3.connect(outFile) cursor = connection.cursor() - #creation de la table - schema = "CREATE TABLE \"trajectories\"(trajectory_id INTEGER,frame_number INTEGER, x_coordinate REAL, y_coordinate REAL, PRIMARY KEY(trajectory_id, frame_number))" + schema = "CREATE TABLE \"positions\"(trajectory_id INTEGER,frame_number INTEGER, x_coordinate REAL, y_coordinate REAL, PRIMARY KEY(trajectory_id, frame_number))" cursor.execute(schema) + trajectory_id = 0 frame_number = 0 if trajectoryType == 'feature': @@ -37,16 +37,12 @@ frame_number = 0 for position in trajectory.getPositions(): frame_number += 1 - requete = "insert into trajectories(trajectory_id, frame_number, x_coordinate, y_coordinate) values (?,?,?,?)" - cursor.execute(requete,(trajectory_id,frame_number,position.x,position.y)) + query = "insert into positions (trajectory_id, frame_number, x_coordinate, y_coordinate) values (?,?,?,?)" + cursor.execute(query,(trajectory_id,frame_number,position.x,position.y)) connection.commit() connection.close() -def testWrite(): - features = loadTrajectoriesFromSqlite("/home/francois/Unison/École/12Été/CRSNG/Data/prototypes.sqlite",'feature',-1) - writeTrajectoriesToSqlite(features, "/home/francois/Unison/École/12Été/CRSNG/TAAM-Experiments/resultats/testWrite.sqlite", 'feature') - def loadPrototypeMatchIndexesFromSqlite(filename): """ This function loads the prototypes table in the database of name <filename>. @@ -70,6 +66,7 @@ return matched_indexes def testloadPrototypeMatchIndexesFromSqlite(): + 'TODO: write as doctest' empty_list = loadPrototypeMatchIndexesFromSqlite("bidon") if empty_list == []: print "Empty list test Ok"