@@ -2164,20 +2164,17 @@ def test_dataset_info(self):
21642164 dataset = self ._api .dataset_info (repo_id = DUMMY_DATASET_ID )
21652165 assert isinstance (dataset .card_data , DatasetCardData ) and len (dataset .card_data ) > 0
21662166 assert isinstance (dataset .siblings , list ) and len (dataset .siblings ) > 0
2167- self . assertIsInstance (dataset , DatasetInfo )
2168- self . assertNotEqual ( dataset .sha , DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT )
2167+ assert isinstance (dataset , DatasetInfo )
2168+ assert dataset .sha != DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT
21692169 dataset = self ._api .dataset_info (
21702170 repo_id = DUMMY_DATASET_ID ,
21712171 revision = DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT ,
21722172 )
2173- self . assertIsInstance (dataset , DatasetInfo )
2174- self . assertEqual ( dataset .sha , DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT )
2173+ assert isinstance (dataset , DatasetInfo )
2174+ assert dataset .sha == DUMMY_DATASET_ID_REVISION_ONE_SPECIFIC_COMMIT
21752175
21762176 def test_dataset_info_with_file_metadata (self ):
2177- dataset = self ._api .dataset_info (
2178- repo_id = SAMPLE_DATASET_IDENTIFIER ,
2179- files_metadata = True ,
2180- )
2177+ dataset = self ._api .dataset_info (repo_id = SAMPLE_DATASET_IDENTIFIER , files_metadata = True )
21812178 files = dataset .siblings
21822179 assert files is not None
21832180 self ._check_siblings_metadata (files )
0 commit comments