File tree Expand file tree Collapse file tree 1 file changed +34
-23
lines changed Expand file tree Collapse file tree 1 file changed +34
-23
lines changed Original file line number Diff line number Diff line change 2
2
3
3
import deepstack .core as ds
4
4
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" ]
24
36
25
- MOCK_PREDICTIONS = MOCK_RESPONSE ['predictions' ]
26
37
27
38
def test_get_object_labels ():
28
39
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
You can’t perform that action at this time.
0 commit comments