[youtube] Update tests

[yt-dlp.git] / test / helper.py

diff --git a/test/helper.py b/test/helper.py
index 598f803f48513d881343c4034c5eb1f4695b2efa..c9293f81aef6a2ecaf0a3a3000773da292c383e0 100644 (file)
--- a/test/helper.py
+++ b/test/helper.py
@@ -224,6 +224,8 @@ def sanitize(key, value):
             return f'md5:{md5(value)}'
         elif isinstance(value, list) and len(value) > 10:
             return f'count:{len(value)}'
+        elif key.endswith('_count') and isinstance(value, int):
+            return int
         return value
 
     test_info_dict = {
@@ -259,6 +261,8 @@ def expect_info_dict(self, got_dict, expected_dict):
         def _repr(v):
             if isinstance(v, compat_str):
                 return "'%s'" % v.replace('\\', '\\\\').replace("'", "\\'").replace('\n', '\\n')
+            elif isinstance(v, type):
+                return v.__name__
             else:
                 return repr(v)
         info_dict_str = ''