diff --git a/tests/test_main.py b/tests/test_main.py index e159747..c15dae7 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -5,17 +5,16 @@ import sys import os -from unittest.mock import patch, MagicMock -from datetime import datetime -import pytest +from unittest.mock import patch +import pytest # pylint: disable=import-error # Add parent directory to path for imports sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) -import main +import main # pylint: disable=import-error,wrong-import-position -class TestWorkoutData: +class TestWorkoutData: # pylint: disable=too-few-public-methods """Test workout data structures and constants.""" def test_workout_groups_exist(self): @@ -291,7 +290,7 @@ def test_stats_command(self, mock_input, mock_print): assert any("completed" in str(call).lower() for call in printed_output) -class TestVideoFunctionality: +class TestVideoFunctionality: # pylint: disable=too-few-public-methods """Test video playback functionality.""" @patch("subprocess.call")