Skip to content

Commit 0602bba

Browse files
committed
Minor PEP8 fixes
1 parent 63f0b8c commit 0602bba

File tree

6 files changed

+32
-15
lines changed

6 files changed

+32
-15
lines changed

tests/hubstorage/test_collections.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,8 @@
1313

1414

1515
def _mkitem():
16-
return dict(field1='value1', field2=['value2a', 'value2b'], field3=3, field4={'v4k': 'v4v'})
16+
return dict(field1='value1', field2=['value2a', 'value2b'],
17+
field3=3, field4={'v4k': 'v4v'})
1718

1819

1920
def test_simple_count(hsproject):
@@ -73,7 +74,8 @@ def post_scan_test(hsproject):
7374
assert len(result) == 11
7475

7576
# combining with normal filters
76-
result = list(col.get(filter='["counter", ">", [5]]', prefix='post_scan_test1'))
77+
result = list(col.get(filter='["counter", ">", [5]]',
78+
prefix='post_scan_test1'))
7779
# 10-19
7880
assert len(result) == 10
7981

tests/hubstorage/test_frontier.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -31,10 +31,12 @@ def test_add_multiple_chunks(hsproject):
3131
fps1 = [{'fp': '/index_%s.html' % fp} for fp in range(0, batch_size)]
3232
frontier.add(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, fps1)
3333

34-
fps2 = [{'fp': '/index_%s.html' % fp} for fp in range(batch_size, batch_size * 2)]
34+
fps2 = [{'fp': '/index_%s.html' % fp}
35+
for fp in range(batch_size, batch_size * 2)]
3536
frontier.add(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, fps2)
3637

37-
fps3 = [{'fp': '/index_%s.html' % fp} for fp in range(batch_size * 2, batch_size * 3)]
38+
fps3 = [{'fp': '/index_%s.html' % fp}
39+
for fp in range(batch_size * 2, batch_size * 3)]
3840
frontier.add(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, fps3)
3941
frontier.flush()
4042

@@ -49,7 +51,8 @@ def test_add_multiple_chunks(hsproject):
4951
assert frontier.newcount == 150 + old_count
5052

5153
# get first 100
52-
batches = list(frontier.read(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, mincount=100))
54+
batches = list(frontier.read(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT,
55+
mincount=100))
5356
urls = [_get_urls(batch) for batch in batches]
5457
expected_urls = [[fp['fp'] for fp in fps1 + fps2]]
5558
assert urls == expected_urls
@@ -74,7 +77,8 @@ def test_add_big_chunk(hsproject):
7477
frontier.flush()
7578

7679
# get first 100
77-
batches = list(frontier.read(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, mincount=100))
80+
batches = list(frontier.read(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT,
81+
mincount=100))
7882
urls = [_get_urls(batch) for batch in batches]
7983
expected_urls = [[fp['fp'] for fp in fps1[:100]]]
8084
assert urls == expected_urls
@@ -84,7 +88,8 @@ def test_add_big_chunk(hsproject):
8488
frontier.delete(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, ids)
8589

8690
# get next 100
87-
batches = list(frontier.read(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, mincount=100))
91+
batches = list(frontier.read(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT,
92+
mincount=100))
8893
urls = [_get_urls(batch) for batch in batches]
8994
expected_urls = [[fp['fp'] for fp in fps1[100:200]]]
9095
assert urls == expected_urls
@@ -94,7 +99,8 @@ def test_add_big_chunk(hsproject):
9499
frontier.delete(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, ids)
95100

96101
# get next 100
97-
batches = list(frontier.read(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT, mincount=100))
102+
batches = list(frontier.read(TEST_FRONTIER_NAME, TEST_FRONTIER_SLOT,
103+
mincount=100))
98104
urls = [_get_urls(batch) for batch in batches]
99105
expected_urls = [[fp['fp'] for fp in fps1[200:300]]]
100106
assert urls == expected_urls

tests/hubstorage/test_jobq.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ def test_push_with_unique(hsproject):
8181
jobq.push(TEST_SPIDER_NAME, unique='h2')
8282
jobq.push(TEST_SPIDER_NAME, unique='h1')
8383

84+
8485
def test_startjob(hsproject):
8586
jobq = hsproject.jobq
8687
qj = jobq.push(TEST_SPIDER_NAME)

tests/hubstorage/test_project.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,8 @@ def test_auth(hsclient):
115115
raise AssertionError('401 not raised')
116116

117117
try:
118-
hsc.get_project(TEST_PROJECT_ID).get_job((TEST_PROJECT_ID, 1, 1)).items.list()
118+
hsc.get_project(TEST_PROJECT_ID).get_job(
119+
(TEST_PROJECT_ID, 1, 1)).items.list()
119120
except HTTPError as exc:
120121
assert exc.response.status_code == 401
121122
else:
@@ -179,13 +180,13 @@ def test_requests(hsproject):
179180
job = hsproject.push_job(TEST_SPIDER_NAME, state='running')
180181
# top parent
181182
r1 = job.requests.add(url='http://test.com/', status=200, method='GET',
182-
rs=1337, duration=5, parent=None, ts=ts)
183+
rs=1337, duration=5, parent=None, ts=ts)
183184
# first child
184185
r2 = job.requests.add(url='http://test.com/2', status=400, method='POST',
185-
rs=0, duration=1, parent=r1, ts=ts + 1)
186+
rs=0, duration=1, parent=r1, ts=ts + 1)
186187
# another child with fingerprint set
187188
r3 = job.requests.add(url='http://test.com/3', status=400, method='PUT',
188-
rs=0, duration=1, parent=r1, ts=ts + 2, fp='1234')
189+
rs=0, duration=1, parent=r1, ts=ts + 2, fp='1234')
189190

190191
job.requests.close()
191192
rr = job.requests.list()
@@ -243,7 +244,7 @@ def test_samples(hsproject):
243244
def test_jobsummary(hsproject):
244245
js = hsproject.jobsummary()
245246
assert js.get('project') == int(hsproject.projectid), js
246-
assert js.get('has_capacity') == True, js
247+
assert js.get('has_capacity') is True, js
247248
assert 'pending' in js, js
248249
assert 'running' in js, js
249250

@@ -253,8 +254,7 @@ def test_bulkdata(hsproject):
253254
for i in range(20):
254255
j.logs.info("log line %d" % i)
255256
j.items.write(dict(field1="item%d" % i))
256-
j.requests.add("http://test.com/%d" % i,
257-
200, 'GET', 10, None, 10, 120)
257+
j.requests.add("http://test.com/%d" % i, 200, 'GET', 10, None, 10, 120)
258258
for resourcename in ('logs', 'items', 'requests'):
259259
resource = getattr(j, resourcename)
260260
resource.flush()

tests/hubstorage/test_retry.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,7 @@ def request_callback(request):
153153
assert dict(job_metadata) == dict(job.metadata)
154154
assert attempts_count[0] == 4
155155

156+
156157
@responses.activate
157158
def test_api_delete_can_be_set_to_non_idempotent(hsspiderid):
158159
# Prepare
@@ -181,6 +182,7 @@ def test_api_delete_can_be_set_to_non_idempotent(hsspiderid):
181182
assert attempts_count_delete[0] == 1
182183
assert err is not None
183184

185+
184186
@responses.activate
185187
def test_collection_store_and_delete_are_retried():
186188
# Prepare
@@ -202,6 +204,7 @@ def test_collection_store_and_delete_are_retried():
202204
assert attempts_count_post[0] == 3
203205
assert attempts_count_delete[0] == 3
204206

207+
205208
@responses.activate
206209
def test_delete_requests_are_retried(hsspiderid):
207210
# Prepare
@@ -229,6 +232,7 @@ def test_delete_requests_are_retried(hsspiderid):
229232
# Assert
230233
assert attempts_count_delete[0] == 3
231234

235+
232236
@responses.activate
233237
def test_metadata_save_does_retry(hsspiderid):
234238
# Prepare
@@ -252,6 +256,7 @@ def test_metadata_save_does_retry(hsspiderid):
252256
# Assert
253257
assert attempts_count_post[0] == 3
254258

259+
255260
@responses.activate
256261
def test_push_job_does_not_retry():
257262
# Prepare
@@ -273,6 +278,7 @@ def test_push_job_does_not_retry():
273278
assert err.response.status_code == 504
274279
assert attempts_count[0] == 1
275280

281+
276282
@responses.activate
277283
def test_get_job_does_retry(hsspiderid):
278284
# Prepare
@@ -321,6 +327,7 @@ def test_get_job_does_fails_if_no_retries(hsspiderid):
321327
assert err.response.status_code == 504
322328
assert attempts_count[0] == 1
323329

330+
324331
@responses.activate
325332
def test_get_job_does_fails_on_too_many_retries(hsspiderid):
326333
# Prepare

tests/hubstorage/test_system.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,7 @@ def _do_test_job(hsproject, panelproject,
7575
assert job.metadata.get('close_reason') == expected_close_reason
7676
return job
7777

78+
7879
def _run_runner(hsproject, pushed, close_reason):
7980
client = HubstorageClient(endpoint=TEST_ENDPOINT, auth=TEST_AUTH)
8081
with closing(client) as runnerclient:

0 commit comments

Comments
 (0)