diff --git a/tests/download/test_sentinelhub_statistical_client.py b/tests/download/test_sentinelhub_statistical_client.py index d77e7ad4..ca89f32e 100644 --- a/tests/download/test_sentinelhub_statistical_client.py +++ b/tests/download/test_sentinelhub_statistical_client.py @@ -45,9 +45,9 @@ def test_statistical_client_download_per_interval(download_request: DownloadRequ ], ) - data = client.download(download_request) + data = client.download([download_request]) - assert data == { + assert data[0] == { "data": [ {"interval": {"from": "2020-01-05", "to": "2020-01-05"}, "outputs": 0}, {"interval": {"from": "2020-01-10", "to": "2020-01-10"}, "error": {"type": "BAD_REQUEST"}}, @@ -80,5 +80,5 @@ def test_statistical_client_runs_out_of_retries(download_request: DownloadReques ) with pytest.raises(DownloadFailedException) as exception_info: - client.download(download_request) + client.download([download_request]) assert str(exception_info.value) == "No more interval retries available, download unsuccessful"