diff --git a/test/test_download.py b/test/test_download.py index 61015562da..5c5824d65a 100644 --- a/test/test_download.py +++ b/test/test_download.py @@ -200,8 +200,9 @@ def try_rm_tcs_files(tcs=None): self.assertEqual( test_case['playlist_duration_sum'], got_duration) - for tc in test_cases: - expect_value(self, res_dict['id'], tc['info_dict']['id'], 'id') + for tc_num, tc in enumerate(test_cases): + tc_res_dict = res_dict['entries'][tc_num] if is_playlist else res_dict + expect_info_dict(self, tc_res_dict, tc.get('info_dict', {})) tc_filename = get_tc_filename(tc) if not test_case.get('params', {}).get('skip_download', False): self.assertTrue(os.path.exists(tc_filename), msg='Missing file ' + tc_filename)