Merge pull request #1246 from Mause/variable-user-agent

test: handle user agent including python version

Author: Mause

duckdb_engine/tests/test_basic.py

@@ -525,6 +525,9 @@ def test_url_config_and_dict_config() -> None:
         assert memory_limit in ("500.0MB", "476.8 MiB")
 
 
+user_agent_re = r"duckdb/.*(.*) python(/.*)? duckdb_engine/.*(sqlalchemy/.*)"
+
+
 @mark.skipif(
     supports_user_agent is False,
     reason="custom_user_agent is not supported for DuckDB version < 0.9.2",
@@ -536,7 +539,7 @@ def test_user_agent() -> None:
         res = conn.execute(text("PRAGMA USER_AGENT"))
         row = res.first()
         assert row is not None
-        assert re.match(r"duckdb/.*(.*) python duckdb_engine/.*(sqlalchemy/.*)", row[0])
+        assert re.match(user_agent_re, row[0])
 
 
 @mark.skipif(
@@ -551,9 +554,7 @@ def test_user_agent_with_custom_user_agent() -> None:
         res = conn.execute(text("PRAGMA USER_AGENT"))
         row = res.first()
         assert row is not None
-        assert re.match(
-            r"duckdb/.*(.*) python duckdb_engine/.*(sqlalchemy/.*) custom", row[0]
-        )
+        assert re.match(user_agent_re + " custom", row[0])
 
     # Check that connect hasn't mutated the caller's data.
     assert connect_args["config"]["custom_user_agent"] == "custom"