From fdf701e8f693e46c3ae854634e0460db58a2a638 Mon Sep 17 00:00:00 2001 From: coletdjnz Date: Sun, 18 May 2025 11:49:45 +1200 Subject: [PATCH] clean up conftest --- test/test_pot/conftest.py | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/test/test_pot/conftest.py b/test/test_pot/conftest.py index f0f24d4051..ff0667e928 100644 --- a/test/test_pot/conftest.py +++ b/test/test_pot/conftest.py @@ -1,3 +1,5 @@ +import collections + import pytest from yt_dlp import YoutubeDL @@ -14,27 +16,22 @@ class MockLogger(IEContentProviderLogger): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) - self.messages = {} + self.messages = collections.defaultdict(list) def trace(self, message: str): - self.messages.setdefault('trace', []).append(message) - pass + self.messages['trace'].append(message) def debug(self, message: str): - self.messages.setdefault('debug', []).append(message) - pass + self.messages['debug'].append(message) def info(self, message: str): - self.messages.setdefault('info', []).append(message) - pass + self.messages['info'].append(message) def warning(self, message: str, *, once=False): - self.messages.setdefault('warning', []).append(message) - pass + self.messages['warning'].append(message) def error(self, message: str): - self.messages.setdefault('error', []).append(message) - pass + self.messages['error'].append(message) @pytest.fixture