Skip to content

Commit c214086

Browse files
committed
black format
1 parent 23a41dc commit c214086

File tree

1 file changed

+34
-23
lines changed

1 file changed

+34
-23
lines changed

tests/test_deepstack.py

Lines changed: 34 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -2,30 +2,41 @@
22

33
import deepstack.core as ds
44

5-
MOCK_RESPONSE ={'success': True,
6-
'predictions': [{'confidence': 0.6998661,
7-
'label': 'person',
8-
'y_min': 0,
9-
'x_min': 258,
10-
'y_max': 676,
11-
'x_max': 485},
12-
{'confidence': 0.7996547,
13-
'label': 'person',
14-
'y_min': 0,
15-
'x_min': 405,
16-
'y_max': 652,
17-
'x_max': 639},
18-
{'confidence': 0.59745613,
19-
'label': 'dog',
20-
'y_min': 311,
21-
'x_min': 624,
22-
'y_max': 591,
23-
'x_max': 825}]}
5+
MOCK_RESPONSE = {
6+
"success": True,
7+
"predictions": [
8+
{
9+
"confidence": 0.6998661,
10+
"label": "person",
11+
"y_min": 0,
12+
"x_min": 258,
13+
"y_max": 676,
14+
"x_max": 485,
15+
},
16+
{
17+
"confidence": 0.7996547,
18+
"label": "person",
19+
"y_min": 0,
20+
"x_min": 405,
21+
"y_max": 652,
22+
"x_max": 639,
23+
},
24+
{
25+
"confidence": 0.59745613,
26+
"label": "dog",
27+
"y_min": 311,
28+
"x_min": 624,
29+
"y_max": 591,
30+
"x_max": 825,
31+
},
32+
],
33+
}
34+
35+
MOCK_PREDICTIONS = MOCK_RESPONSE["predictions"]
2436

25-
MOCK_PREDICTIONS = MOCK_RESPONSE['predictions']
2637

2738
def test_get_object_labels():
2839
object_labels = ds.get_object_labels(MOCK_PREDICTIONS)
29-
assert 'dog' in object_labels
30-
assert 'person' in object_labels
31-
assert len(object_labels) == 2
40+
assert "dog" in object_labels
41+
assert "person" in object_labels
42+
assert len(object_labels) == 2

0 commit comments

Comments
 (0)