test_video_processor_mediapipe.py

import json
import logging
from app.video_processor import extract_motion_and_frame
from app.mediapipe_processor import extract_face_signals_facelandmarker
from app.pose_processor import extract_body_signals
from app.config import VIDEO_TEMP_PATH, JSON_OUTPUT_PATH
from app.logger import log
import os
logger = logging.getLogger("test_logger")

def test_extract_face_signals():
    video_path = str(VIDEO_TEMP_PATH)
    signals = extract_face_signals_facelandmarker(video_path)
    log(f"Face signals extracted: {signals}", logger=logger)

def test_extract_body_signals():
    video_path = str(VIDEO_TEMP_PATH)
    signals = extract_body_signals(video_path)
    log(f"Body signals extracted: {signals}", logger=logger)

def test_mediapipe_runner():
    # Simulate running the mediapipe_runner
    os.system(f"python -m app.mediapipe_runner {str(VIDEO_TEMP_PATH)} > {str(JSON_OUTPUT_PATH)}")
    
        # Load the JSON output and check its structure
    with open(JSON_OUTPUT_PATH, 'r') as f:
        data = json.load(f)
    
    log(f"Mediapipe runner output: {data}", logger=logger)

def test_extract_motion_and_frame():
    video_path = str(VIDEO_TEMP_PATH)
    motion = extract_motion_and_frame(video_path)
    log(f"Motion extracted: {motion}", logger=logger)

if __name__ == "__main__":
    #test_extract_face_signals()
    test_extract_body_signals()
    #test_extract_motion_and_frame()
    #test_mediapipe_runner()