Fixups for #9

elprans · elprans · commit b91511f70ea4 · 2023-06-22T15:21:33.000-07:00
diff --git a/_python/pgbench_python.py b/_python/pgbench_python.py
@@ -201,7 +201,7 @@ async def async_psycopg_copy(conn, query, args):
     for row in rows:
         writer.writerow(row)
     f.seek(0)
-    
+
     async with conn.cursor() as cur:
         async with cur.copy(query) as copy:
             await copy.write(f.read())
@@ -442,6 +442,7 @@ def die(msg):
             'psycopg3-async',
             'postgresql'
         ],
+    )
     parser.add_argument(
         'queryfile', help='file to read benchmark query information from')
 
@@ -499,7 +500,7 @@ def die(msg):
         arg_format = 'native'
     elif args.driver == 'psycopg2':
         connector, executor, copy_executor, batch_executor = (
-            psycopg2_connect, psycopg2_execute, 
+            psycopg2_connect, psycopg2_execute,
             psycopg2_copy, psycopg2_executemany,
         )
         is_async = False
@@ -511,7 +512,7 @@ def die(msg):
         arg_format = 'python'
     elif args.driver == 'psycopg3-async':
         connector, executor, copy_executor, batch_executor = (
-            async_psycopg_connect, async_psycopg_execute, 
+            async_psycopg_connect, async_psycopg_execute,
             async_psycopg_copy, async_psycopg_executemany,
         )
         is_async = True
diff --git a/pgbench b/pgbench
@@ -360,6 +360,7 @@ BENCHMARKS = [
     'python-aiopg',
     'python-aiopg-tuples',
     'python-asyncpg',
+    'python-psycopg3-async',
     'nodejs-pg-js',
     'nodejs-ts-postgres'
 ]

Original file line number	Diff line number	Diff line change
`@@ -360,6 +360,7 @@ BENCHMARKS = [`
`360`	`360`	`'python-aiopg',`
`361`	`361`	`'python-aiopg-tuples',`
`362`	`362`	`'python-asyncpg',`
	`363`	`+ 'python-psycopg3-async',`
`363`	`364`	`'nodejs-pg-js',`
`364`	`365`	`'nodejs-ts-postgres'`
`365`	`366`	`]`