@@ -40,9 +40,7 @@ def adbc_postgresql_null_session(postgres_service: PostgresService) -> Generator
40
40
@xfail_if_driver_missing
41
41
def test_postgresql_single_null_parameter (adbc_postgresql_null_session : AdbcDriver ) -> None :
42
42
"""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 )
46
44
assert result .rows_affected in (- 1 , 1 )
47
45
48
46
# Verify data
@@ -56,8 +54,7 @@ def test_postgresql_single_null_parameter(adbc_postgresql_null_session: AdbcDriv
56
54
def test_postgresql_all_null_parameters (adbc_postgresql_null_session : AdbcDriver ) -> None :
57
55
"""Test executing with all NULL parameters."""
58
56
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
61
58
)
62
59
assert result .rows_affected in (- 1 , 1 )
63
60
@@ -74,8 +71,7 @@ def test_postgresql_all_null_parameters(adbc_postgresql_null_session: AdbcDriver
74
71
def test_postgresql_mixed_null_parameters (adbc_postgresql_null_session : AdbcDriver ) -> None :
75
72
"""Test executing with mixed NULL and non-NULL parameters."""
76
73
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"
79
75
)
80
76
assert result .rows_affected in (- 1 , 1 )
81
77
@@ -99,15 +95,12 @@ def test_postgresql_execute_many_with_nulls(adbc_postgresql_null_session: AdbcDr
99
95
]
100
96
101
97
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
104
99
)
105
100
assert result .rows_affected in (- 1 , 4 , 1 )
106
101
107
102
# 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" )
111
104
assert len (result .data ) == 4
112
105
113
106
# Check each row
0 commit comments