Switched to getuser for tests

Author: ankane

tests/test_pg8000.py

@@ -1,10 +1,10 @@
+from getpass import getuser
 import numpy as np
-import os
 from pgvector import HalfVector, SparseVector, Vector
 from pgvector.pg8000 import register_vector
 from pg8000.native import Connection
 
-conn = Connection(os.environ["USER"], database='pgvector_python_test')
+conn = Connection(getuser(), database='pgvector_python_test')
 
 conn.run('CREATE EXTENSION IF NOT EXISTS vector')
 conn.run('DROP TABLE IF EXISTS pg8000_items')

tests/test_sqlalchemy.py

@@ -1,6 +1,6 @@
 import asyncpg
+from getpass import getuser
 import numpy as np
-import os
 from pgvector import HalfVector, SparseVector, Vector
 from pgvector.sqlalchemy import VECTOR, HALFVEC, BIT, SPARSEVEC, avg, sum
 import pytest
@@ -28,7 +28,7 @@ def psycopg2_connect(dbapi_connection, connection_record):
     register_vector(dbapi_connection)
 
 
-pg8000_engine = create_engine(f'postgresql+pg8000://{os.environ["USER"]}@localhost/pgvector_python_test')
+pg8000_engine = create_engine(f'postgresql+pg8000://{getuser()}@localhost/pgvector_python_test')
 
 if sqlalchemy_version > 1:
     psycopg_engine = create_engine('postgresql+psycopg://localhost/pgvector_python_test')