1
1
import pytest
2
2
3
- from helpers import (
3
+ from . helpers import (
4
4
TEST_SLICE_NAME ,
5
5
TEST_DATASET_NAME ,
6
6
TEST_IMG_URLS ,
17
17
DATASET_ID_KEY ,
18
18
)
19
19
20
+
20
21
@pytest .fixture ()
21
22
def dataset (CLIENT ):
22
23
ds = CLIENT .create_dataset (TEST_DATASET_NAME )
@@ -25,6 +26,7 @@ def dataset(CLIENT):
25
26
response = CLIENT .delete_dataset (ds .id )
26
27
assert response == {}
27
28
29
+
28
30
def test_dataset_create_and_delete (CLIENT ):
29
31
# Creation
30
32
ds = CLIENT .create_dataset (TEST_DATASET_NAME )
@@ -88,14 +90,17 @@ def test_dataset_list_autotags(CLIENT, dataset):
88
90
autotag_response = CLIENT .list_autotags (dataset .id )
89
91
assert autotag_response == []
90
92
93
+
91
94
def test_slice_create_and_delete (dataset ):
92
95
# Dataset upload
93
96
ds_items = []
94
97
for url in TEST_IMG_URLS :
95
- ds_items .append (DatasetItem (
96
- image_location = url ,
97
- reference_id = reference_id_from_url (url ),
98
- ))
98
+ ds_items .append (
99
+ DatasetItem (
100
+ image_location = url ,
101
+ reference_id = reference_id_from_url (url ),
102
+ )
103
+ )
99
104
response = dataset .append (ds_items )
100
105
assert ERROR_PAYLOAD not in response .json ()
101
106
@@ -121,10 +126,12 @@ def test_slice_append(dataset):
121
126
# Dataset upload
122
127
ds_items = []
123
128
for url in TEST_IMG_URLS :
124
- ds_items .append (DatasetItem (
125
- image_location = url ,
126
- reference_id = reference_id_from_url (url ),
127
- ))
129
+ ds_items .append (
130
+ DatasetItem (
131
+ image_location = url ,
132
+ reference_id = reference_id_from_url (url ),
133
+ )
134
+ )
128
135
response = dataset .append (ds_items )
129
136
assert ERROR_PAYLOAD not in response .json ()
130
137
0 commit comments