test_db.py

import psycopg
import os
from dotenv import load_dotenv

load_dotenv()
db_url = os.environ.get("DATABASE_URL")

conn = psycopg.connect(db_url)
print("Connected.", conn.closed)

with conn:
    print("Inside with block")
    
print("After with block, closed:", conn.closed)