diff --git a/tests/test_hf_api.py b/tests/test_hf_api.py index f121ae4b0f..ff6076b7a3 100644 --- a/tests/test_hf_api.py +++ b/tests/test_hf_api.py @@ -2164,20 +2164,17 @@ def test_dataset_info(self): dataset = self._api.dataset_info(repo_id=DUMMY_DATASET_ID) assert isinstance(dataset.card_data, DatasetCardData) and len(dataset.card_data) > 0 assert isinstance(dataset.siblings, list) and len(dataset.siblings) > 0 - self.assertIsInstance(dataset, DatasetInfo) - self.assertNotEqual(dataset.sha, DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT) + assert isinstance(dataset, DatasetInfo) + assert dataset.sha != DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT dataset = self._api.dataset_info( repo_id=DUMMY_DATASET_ID, revision=DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT, ) - self.assertIsInstance(dataset, DatasetInfo) - self.assertEqual(dataset.sha, DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT) + assert isinstance(dataset, DatasetInfo) + assert dataset.sha == DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT def test_dataset_info_with_file_metadata(self): - dataset = self._api.dataset_info( - repo_id=SAMPLE_DATASET_IDENTIFIER, - files_metadata=True, - ) + dataset = self._api.dataset_info(repo_id=SAMPLE_DATASET_IDENTIFIER, files_metadata=True) files = dataset.siblings assert files is not None self._check_siblings_metadata(files) diff --git a/tests/testing_utils.py b/tests/testing_utils.py index 10bdf93263..a3e91fbbdb 100644 --- a/tests/testing_utils.py +++ b/tests/testing_utils.py @@ -47,8 +47,8 @@ SAMPLE_DATASET_IDENTIFIER = "lhoestq/custom_squad" # Example dataset ids -DUMMY_DATASET_ID = "lhoestq/test" -DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT = "81d06f998585f8ee10e6e3a2ea47203dc75f2a16" # on branch "test-branch" +DUMMY_DATASET_ID = "gaia-benchmark/GAIA" +DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT = "c603981e170e9e333934a39781d2ae3a2677e81f" # on branch "test-branch" YES = ("y", "yes", "t", "true", "on", "1") NO = ("n", "no", "f", "false", "off", "0")