test(modules): Adds postgres test ensure both drivers are compatible (#828)

Once the new docs site is up I will likely add a demo there, but for the
moment a quick basic unit test to ensure both are properly working.

Author: terry-docker

modules/postgres/tests/test_postgres.py

@@ -133,3 +133,21 @@ def test_none_driver_urls():
 
         url = container.get_connection_url(driver=None)
         assert url == expected_url
+
+
+def test_psycopg_versions():
+    """Test that both psycopg2 and psycopg (v2 and v3) work with the container."""
+
+    postgres_container = PostgresContainer("postgres:16-alpine", driver="psycopg2")
+    with postgres_container as postgres:
+        engine = sqlalchemy.create_engine(postgres.get_connection_url())
+        with engine.begin() as connection:
+            result = connection.execute(sqlalchemy.text("SELECT 1 as test"))
+            assert result.scalar() == 1
+
+    postgres_container = PostgresContainer("postgres:16-alpine", driver="psycopg")
+    with postgres_container as postgres:
+        engine = sqlalchemy.create_engine(postgres.get_connection_url())
+        with engine.begin() as connection:
+            result = connection.execute(sqlalchemy.text("SELECT 1 as test"))
+            assert result.scalar() == 1