@@ -40,9 +40,7 @@ def adbc_postgresql_null_session(postgres_service: PostgresService) -> Generator
4040@xfail_if_driver_missing
4141def test_postgresql_single_null_parameter (adbc_postgresql_null_session : AdbcDriver ) -> None :
4242 """Test executing with a single NULL parameter."""
43- result = adbc_postgresql_null_session .execute (
44- "INSERT INTO test_null_params (col1) VALUES ($1)" , None
45- )
43+ result = adbc_postgresql_null_session .execute ("INSERT INTO test_null_params (col1) VALUES ($1)" , None )
4644 assert result .rows_affected in (- 1 , 1 )
4745
4846 # Verify data
@@ -56,8 +54,7 @@ def test_postgresql_single_null_parameter(adbc_postgresql_null_session: AdbcDriv
5654def test_postgresql_all_null_parameters (adbc_postgresql_null_session : AdbcDriver ) -> None :
5755 """Test executing with all NULL parameters."""
5856 result = adbc_postgresql_null_session .execute (
59- "INSERT INTO test_null_params (col1, col2, col3) VALUES ($1, $2, $3)" ,
60- None , None , None
57+ "INSERT INTO test_null_params (col1, col2, col3) VALUES ($1, $2, $3)" , None , None , None
6158 )
6259 assert result .rows_affected in (- 1 , 1 )
6360
@@ -74,8 +71,7 @@ def test_postgresql_all_null_parameters(adbc_postgresql_null_session: AdbcDriver
7471def test_postgresql_mixed_null_parameters (adbc_postgresql_null_session : AdbcDriver ) -> None :
7572 """Test executing with mixed NULL and non-NULL parameters."""
7673 result = adbc_postgresql_null_session .execute (
77- "INSERT INTO test_null_params (col1, col2, col3) VALUES ($1, $2, $3)" ,
78- "value1" , None , "value3"
74+ "INSERT INTO test_null_params (col1, col2, col3) VALUES ($1, $2, $3)" , "value1" , None , "value3"
7975 )
8076 assert result .rows_affected in (- 1 , 1 )
8177
@@ -99,15 +95,12 @@ def test_postgresql_execute_many_with_nulls(adbc_postgresql_null_session: AdbcDr
9995 ]
10096
10197 result = adbc_postgresql_null_session .execute_many (
102- "INSERT INTO test_null_params (col1, col2, col3) VALUES ($1, $2, $3)" ,
103- parameters
98+ "INSERT INTO test_null_params (col1, col2, col3) VALUES ($1, $2, $3)" , parameters
10499 )
105100 assert result .rows_affected in (- 1 , 4 , 1 )
106101
107102 # Verify data
108- result = adbc_postgresql_null_session .execute (
109- "SELECT * FROM test_null_params ORDER BY id"
110- )
103+ result = adbc_postgresql_null_session .execute ("SELECT * FROM test_null_params ORDER BY id" )
111104 assert len (result .data ) == 4
112105
113106 # Check each row
0 commit comments