1
1
# Placeholder
2
2
3
3
import deepstack .core as ds
4
+ import requests_mock
5
+
6
+ MOCK_IP_ADDRESS = "localhost"
7
+ MOCK_PORT = 5000
8
+ MOCK_URL = "http://{}:{}/v1/vision/detection" .format (MOCK_IP_ADDRESS , MOCK_PORT )
9
+
10
+ MOCK_BYTES = b"Test"
11
+ MOCK_API_KEY = "mock_api_key"
12
+ MOCK_TIMEOUT = 8
4
13
5
14
MOCK_RESPONSE = {
6
15
"success" : True ,
37
46
CONFIDENCE_THRESHOLD = 0.7
38
47
39
48
49
+ def test_DeepstackObject ():
50
+ with requests_mock .Mocker () as mock_req :
51
+ mock_req .post (MOCK_URL , status_code = ds .HTTP_OK , json = MOCK_RESPONSE )
52
+
53
+ dsobject = ds .DeepstackObject (MOCK_IP_ADDRESS , MOCK_PORT )
54
+ dsobject .process_image_bytes (MOCK_BYTES )
55
+ assert dsobject .predictions == MOCK_PREDICTIONS
56
+
57
+
40
58
def test_get_object_labels ():
41
59
"""Cant always be sure order of returned list items."""
42
60
object_labels = ds .get_object_labels (MOCK_PREDICTIONS )
@@ -45,14 +63,19 @@ def test_get_object_labels():
45
63
assert "person" in object_labels
46
64
assert len (object_labels ) == 2
47
65
66
+
48
67
def test_get_objects_summary ():
49
68
objects_summary = ds .get_objects_summary (MOCK_PREDICTIONS )
50
- assert objects_summary == {'dog' : 1 , 'person' : 2 }
69
+ assert objects_summary == {"dog" : 1 , "person" : 2 }
70
+
51
71
52
72
def test_get_label_confidences ():
53
- label_confidences = ds .get_label_confidences (MOCK_PREDICTIONS , ' person' )
73
+ label_confidences = ds .get_label_confidences (MOCK_PREDICTIONS , " person" )
54
74
assert label_confidences == MOCK_CONFIDENCES
55
75
56
76
57
77
def test_get_confidences_above_threshold ():
58
- assert len (ds .get_confidences_above_threshold (MOCK_CONFIDENCES , CONFIDENCE_THRESHOLD )) == 1
78
+ assert (
79
+ len (ds .get_confidences_above_threshold (MOCK_CONFIDENCES , CONFIDENCE_THRESHOLD ))
80
+ == 1
81
+ )
0 commit comments