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")