mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-07-04 20:03:12 +02:00
Krzysztof kuhy Rudnicki
3ebb97b283
- Move slow hooks (mypy, pylint, bandit, pytest, prettier) to pre-push stage - Remove redundant autoflake (ruff covers F401/F841) - Fix shellcheck OOM by batching files with xargs -n 40 - Remove tracked .o, .wav, .pyc binaries from git - Move pomodoro wav files to ../testsAndMisc_binaries/ with symlinks - Add *.o, *.so, *.a to .gitignore - Refactor hltb._pick_best_hltb_entry to fix C901/PLR0911/SIM102 - Fix SC2034 warnings in gif_to_square.sh and upgrade.sh - Add disk_cleanup_check.sh script - Various test and code improvements across screen_locker, steam_backlog_enforcer, word_frequency, moviepy_showcase
152 lines
4.3 KiB
Python
152 lines
4.3 KiB
Python
"""Mock moviepy modules for all moviepy_showcase tests.
|
|
|
|
This module-level setup installs mock moviepy packages into sys.modules
|
|
so source modules can be imported without moviepy installed.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import importlib
|
|
import sys
|
|
from typing import Any
|
|
from unittest.mock import MagicMock
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
_H, _W = 1080, 1920
|
|
|
|
|
|
def create_mock_clip(**overrides: float | tuple[int, int]) -> MagicMock:
|
|
"""Return a MagicMock that behaves enough like a moviepy clip."""
|
|
clip = MagicMock()
|
|
clip.duration = overrides.get("duration", 2.0)
|
|
clip.size = overrides.get("size", (_W, _H))
|
|
clip.fps = overrides.get("fps", 30)
|
|
chain = [
|
|
"with_fps",
|
|
"with_duration",
|
|
"with_position",
|
|
"with_opacity",
|
|
"with_mask",
|
|
"with_audio",
|
|
"with_effects",
|
|
"with_background_color",
|
|
"with_speed_scaled",
|
|
"with_section_cut_out",
|
|
"with_effects_on_subclip",
|
|
"with_layer_index",
|
|
"with_volume_scaled",
|
|
"with_start",
|
|
"subclipped",
|
|
"cropped",
|
|
"resized",
|
|
"rotated",
|
|
"image_transform",
|
|
"transform",
|
|
"time_transform",
|
|
"to_ImageClip",
|
|
"to_mask",
|
|
"to_RGB",
|
|
]
|
|
for name in chain:
|
|
getattr(clip, name).return_value = clip
|
|
return clip
|
|
|
|
|
|
# ── Build mock module tree ────────────────────────────────────────
|
|
mock_moviepy = MagicMock()
|
|
|
|
_clip_classes = [
|
|
"VideoClip",
|
|
"ColorClip",
|
|
"TextClip",
|
|
"ImageClip",
|
|
"CompositeVideoClip",
|
|
"VideoFileClip",
|
|
"BitmapClip",
|
|
"DataVideoClip",
|
|
"ImageSequenceClip",
|
|
"AudioClip",
|
|
"AudioArrayClip",
|
|
"CompositeAudioClip",
|
|
]
|
|
|
|
_drawing_shape = (_H, _W)
|
|
|
|
|
|
def _reset_side_effects() -> None:
|
|
"""(Re)apply side_effect / return_value on the shared mock_moviepy."""
|
|
for cls_name in _clip_classes:
|
|
getattr(mock_moviepy, cls_name).side_effect = lambda *_a, **_kw: (
|
|
create_mock_clip()
|
|
)
|
|
mock_moviepy.concatenate_videoclips.side_effect = lambda *_a, **_kw: (
|
|
create_mock_clip()
|
|
)
|
|
mock_moviepy.concatenate_audioclips.side_effect = lambda *_a, **_kw: (
|
|
create_mock_clip()
|
|
)
|
|
mock_moviepy.video.compositing.CompositeVideoClip.clips_array.side_effect = (
|
|
lambda *_a, **_kw: create_mock_clip()
|
|
)
|
|
# Drawing tools must return real numpy arrays (used in numpy ops)
|
|
mock_moviepy.video.tools.drawing.circle.return_value = np.zeros(
|
|
_drawing_shape, dtype=np.float64
|
|
)
|
|
mock_moviepy.video.tools.drawing.color_gradient.return_value = np.zeros(
|
|
_drawing_shape, dtype=np.float64
|
|
)
|
|
mock_moviepy.video.tools.drawing.color_split.return_value = np.zeros(
|
|
_drawing_shape, dtype=np.float64
|
|
)
|
|
|
|
|
|
_reset_side_effects()
|
|
|
|
# ── Install into sys.modules ─────────────────────────────────────
|
|
_module_paths = [
|
|
"moviepy",
|
|
"moviepy.video",
|
|
"moviepy.video.fx",
|
|
"moviepy.video.compositing",
|
|
"moviepy.video.compositing.CompositeVideoClip",
|
|
"moviepy.video.tools",
|
|
"moviepy.video.tools.drawing",
|
|
"moviepy.audio",
|
|
"moviepy.audio.fx",
|
|
]
|
|
|
|
|
|
def _install_moviepy_mocks() -> None:
|
|
"""(Re)install this conftest's moviepy mocks into sys.modules."""
|
|
for _mod in _module_paths:
|
|
parts = _mod.split(".")
|
|
obj: Any = mock_moviepy
|
|
for part in parts[1:]:
|
|
obj = getattr(obj, part)
|
|
sys.modules[_mod] = obj
|
|
|
|
|
|
_install_moviepy_mocks()
|
|
|
|
|
|
_source_modules = [
|
|
"python_pkg.moviepy_showcase.moviepy_showcase",
|
|
"python_pkg.moviepy_showcase._moviepy_clip_types",
|
|
"python_pkg.moviepy_showcase._moviepy_video_effects",
|
|
"python_pkg.moviepy_showcase._moviepy_audio_output",
|
|
]
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _reinstall_moviepy_mocks() -> None:
|
|
"""Ensure our moviepy mocks are active even if another conftest overwrote."""
|
|
_install_moviepy_mocks()
|
|
_reset_side_effects()
|
|
# Re-bind cached ``from moviepy import X`` references in source modules
|
|
# that may point to stale MagicMock children from a prior sys.modules entry.
|
|
for mod_name in _source_modules:
|
|
if mod_name in sys.modules:
|
|
importlib.reload(sys.modules[mod_name])
|