@@ -50,7 +50,7 @@ def test_add_dataset(self):
5050
5151 ge .initialize (ge_instance .address (), credentials = ("admin@localhost" , "adminadmin" ))
5252
53- volumes = ge .volumes ( )
53+ volume = ge .volume_by_name ( "test_data" )
5454
5555 geo_transform = ge .GeoTransform (x_min = 180.0 , y_max = 90.0 , x_pixel_size = 0.1 , y_pixel_size = - 0.1 )
5656
@@ -141,7 +141,7 @@ def test_add_dataset(self):
141141 )
142142
143143 dataset_name = ge .add_dataset (
144- volumes [ 0 ] ,
144+ volume ,
145145 add_dataset_properties ,
146146 geoengine_openapi_client .MetaDataDefinition (
147147 meta_data ,
@@ -160,7 +160,7 @@ def test_add_dataset_with_permissions(self):
160160
161161 ge .initialize (ge_instance .address (), credentials = ("admin@localhost" , "adminadmin" ))
162162
163- volumes = ge .volumes ( )
163+ volume = ge .volume_by_name ( "test_data" )
164164
165165 geo_transform = ge .GeoTransform (x_min = 180.0 , y_max = 90.0 , x_pixel_size = 0.1 , y_pixel_size = - 0.1 )
166166
@@ -246,7 +246,7 @@ def test_add_dataset_with_permissions(self):
246246 permisions = [(REGISTERED_USER_ROLE_ID , Permission .READ )]
247247
248248 dataset_name = ge .add_or_replace_dataset_with_permissions (
249- volumes [ 0 ] ,
249+ volume ,
250250 add_dataset_properties ,
251251 geoengine_openapi_client .MetaDataDefinition (
252252 meta_data ,
@@ -294,7 +294,7 @@ def test_add_dataset_with_permissions(self):
294294 )
295295
296296 dataset_name = ge .add_or_replace_dataset_with_permissions (
297- volumes [ 0 ] ,
297+ volume ,
298298 add_dataset_properties ,
299299 geoengine_openapi_client .MetaDataDefinition (
300300 meta_data ,
@@ -336,7 +336,7 @@ def test_add_dataset_with_permissions(self):
336336 )
337337
338338 dataset_name = ge .add_or_replace_dataset_with_permissions (
339- volumes [ 0 ] ,
339+ volume ,
340340 add_dataset_properties ,
341341 geoengine_openapi_client .MetaDataDefinition (
342342 meta_data ,
0 commit comments