python-postgres
diff --git a/‎postgresql/test/perf_query_io.py‎
Lines changed: 9 additions & 8 deletions b/‎postgresql/test/perf_query_io.py‎
Lines changed: 9 additions & 8 deletions
@@ -3,7 +3,7 @@
 # copyright 2009, James William Pye
 # http://python.projects.postgresql.org
 ##
-# Query I/O: Mass insert and select performance
+# Statement I/O: Mass insert and select performance
 ##
 import os
 import time
@@ -14,19 +14,19 @@ def insertSamples(count, insert_records):
 		(-3, 123, 0xfffffea023, 'some_óäæ_thing', 'varying', 'æ')
 		for x in range(count)
 	]
-
 	gen = time.time()
-	insert_records << recs
+	insert_records.load(recs)
 	fin = time.time()
 	xacttime = fin - gen
 	ats = count / xacttime
-	print >>sys.stderr, \
+	sys.stderr.write(
 		"INSERT Summary,\n " \
 		"inserted tuples: %d\n " \
 		"total time: %f\n " \
 		"average tuples per second: %f\n " %(
 			count, xacttime, ats, 
 		)
+	)
 
 def timeTupleRead(portal):
 	loops = 0
@@ -36,27 +36,28 @@ def timeTupleRead(portal):
 	finalis = time.time()
 	looptime = finalis - genesis
 	ats = loops / looptime
-	print >>sys.stderr, \
+	sys.stderr.write(
 		"SELECT Summary,\n " \
 		"looped/tuples: %d\n " \
 		"looptime: %f\n " \
 		"average tuples per second: %f\n " %(loops, looptime, ats,)
+	)
 
 def main(count):
 	execute('CREATE TEMP TABLE samples '
 		'(i2 int2, i4 int4, i8 int8, t text, v varchar, c char)')
-	insert_records = query(
+	insert_records = prepare(
 		"INSERT INTO samples VALUES ($1, $2, $3, $4, $5, $6)"
 	)
-	select_records = query("SELECT * FROM samples")
+	select_records = prepare("SELECT * FROM samples")
 	try:
 		insertSamples(count, insert_records)
 		timeTupleRead(select_records())	
 	finally:
 		execute("DROP TABLE samples")
 
 def command(args):
-	main(int((args + [15000])[1]))
+	main(int((args + [25000])[1]))
 
 if __name__ == '__main__':
 	command(sys.argv)