add await_ attribute to connection.py; remove unused asyncio import in tests

Author: Eugene Shershen

psqlpy_sqlalchemy/connection.py

@@ -35,6 +35,7 @@ class AsyncAdapt_psqlpy_cursor(AsyncAdapt_dbapi_cursor):
         "_invalidate_schema_cache_asof",
         "_rowcount",
         "_rows",
+        "await_",
     )
 
     _adapt_connection: "AsyncAdapt_psqlpy_connection"

tests/test_connection.py

@@ -431,8 +431,6 @@ def test_execute_update_async_greenlet_fix(self, mock_prepare_execute):
         operation = "UPDATE test_table SET name = :name WHERE id = :id"
         parameters = {"name": "test_update", "id": 1}
 
-        import asyncio
-
         # This should not raise any greenlet-related errors
         self.cursor.execute(operation, parameters)