def it_returns_the_dataset(darwin_client: Client):
endpoint: str = "/datasets"
json_response = [{
"name": "dataset-name-1",
"slug": "dataset-slug-1",
"id": 1,
"num_images": 1,
"num_videos": 0,
"progress": 0,
}]
responses.add(responses.GET,
darwin_client.url + endpoint,
json=json_response,
status=200)
actual_dataset = darwin_client.get_remote_dataset("v7/dataset-slug-1")
expected_dataset = RemoteDataset(
team="v7",
name="dataset-name-1",
slug="dataset-slug-1",
dataset_id=1,
item_count=1,
client=darwin_client,
)
assert_dataset(actual_dataset, expected_dataset)
def it_raises_if_dataset_is_not_found(darwin_client: Client):
endpoint: str = "/datasets"
json_response = [{
"name": "dataset-name-1",
"slug": "dataset-slug-1",
"id": 1,
"num_images": 1,
"num_videos": 0,
"progress": 0,
}]
responses.add(responses.GET,
darwin_client.url + endpoint,
json=json_response,
status=200)
with pytest.raises(NotFound):
darwin_client.get_remote_dataset("v7/dataset-slug-2")
