Skip to content

Commit 938f06f

Browse files
committed
add tests
1 parent f038eb7 commit 938f06f

File tree

2 files changed

+28
-4
lines changed

2 files changed

+28
-4
lines changed

requirements-dev.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
pytest
2-
black
2+
black
3+
requests_mock

tests/test_deepstack.py

Lines changed: 26 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,15 @@
11
# Placeholder
22

33
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
413

514
MOCK_RESPONSE = {
615
"success": True,
@@ -37,6 +46,15 @@
3746
CONFIDENCE_THRESHOLD = 0.7
3847

3948

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+
4058
def test_get_object_labels():
4159
"""Cant always be sure order of returned list items."""
4260
object_labels = ds.get_object_labels(MOCK_PREDICTIONS)
@@ -45,14 +63,19 @@ def test_get_object_labels():
4563
assert "person" in object_labels
4664
assert len(object_labels) == 2
4765

66+
4867
def test_get_objects_summary():
4968
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+
5171

5272
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")
5474
assert label_confidences == MOCK_CONFIDENCES
5575

5676

5777
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

Comments
 (0)