Skip to content

Commit 12d6bf3

Browse files
committed
chore: clean up flake8 issues in tests
1 parent a94754c commit 12d6bf3

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

tests/test_query.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -208,9 +208,9 @@ def test_query_api_options_builder(self):
208208

209209
print(f"\nDEBUG options {vars(options)}")
210210
try:
211-
assert(options.tls_root_certs.decode('utf-8') == self.sample_cert)
212-
assert(options.tls_verify == False)
213-
assert(options.proxy == proxy_name)
211+
assert options.tls_root_certs.decode('utf-8') == self.sample_cert
212+
assert not options.tls_verify
213+
assert options.proxy == proxy_name
214214
finally:
215215
self.remove_cert_file(cert_file)
216216

@@ -235,12 +235,12 @@ def test_query_client_with_options(self):
235235

236236
print(f"\nDEBUG client {vars(client)}")
237237
try:
238-
assert(client._token == token)
239-
assert(client._flight_client_options['tls_root_certs'].decode('utf-8') == self.sample_cert)
240-
assert(client._proxy == proxy_name)
241-
# print(f"DEBUG client._flight_client_options['generic_options'] {dict(client._flight_client_options['generic_options'])['grpc.secondary_user_agent']}")
242-
assert(dict(client._flight_client_options['generic_options'])['grpc.secondary_user_agent'].startswith('influxdb3-python/'))
243-
assert(dict(client._flight_client_options['generic_options'])['grpc.http_proxy'] == proxy_name)
238+
assert client._token == token
239+
assert client._flight_client_options['tls_root_certs'].decode('utf-8') == self.sample_cert
240+
assert client._proxy == proxy_name
241+
fc_opts = client._flight_client_options
242+
assert dict(fc_opts['generic_options'])['grpc.secondary_user_agent'].startswith('influxdb3-python/')
243+
assert dict(fc_opts['generic_options'])['grpc.http_proxy'] == proxy_name
244244
finally:
245245
self.remove_cert_file(cert_file)
246246

@@ -254,17 +254,17 @@ def test_client_with_ssl_args(self):
254254
database="my_db",
255255
token="my_token",
256256
proxy=proxy,
257-
ssl_ca_cert = cert_name,
258-
verify_ssl = False
257+
ssl_ca_cert=cert_name,
258+
verify_ssl=False
259259
)
260260

261261
try:
262262
qapi = local_client._query_api
263263
fc_opts = qapi._flight_client_options
264-
assert(qapi._proxy == proxy)
265-
assert(fc_opts['tls_root_certs'].decode('utf-8') == self.sample_cert)
266-
assert(fc_opts['disable_server_verification'] == True)
267-
assert(dict(fc_opts['generic_options'])['grpc.secondary_user_agent'].startswith('influxdb3-python/'))
268-
assert(dict(fc_opts['generic_options'])['grpc.http_proxy'] == proxy)
264+
assert qapi._proxy == proxy
265+
assert fc_opts['tls_root_certs'].decode('utf-8') == self.sample_cert
266+
assert fc_opts['disable_server_verification']
267+
assert dict(fc_opts['generic_options'])['grpc.secondary_user_agent'].startswith('influxdb3-python/')
268+
assert dict(fc_opts['generic_options'])['grpc.http_proxy'] == proxy
269269
finally:
270270
self.remove_cert_file(cert_name)

0 commit comments

Comments
 (0)