import math
from edge.models import Keypoint, BBox
from edge.geometry import (
    com, torso_angle_deg, head_below_hips, visible_count,
    core_visible, normalize,
)

def _kps(overrides):
    kps = [Keypoint(0, 0, 0.0) for _ in range(17)]
    for i, (x, y, c) in overrides.items():
        kps[i] = Keypoint(x, y, c)
    return kps

def test_com_is_mean_of_shoulders_and_hips():
    kps = _kps({5: (0, 0, 1), 6: (10, 0, 1), 11: (0, 10, 1), 12: (10, 10, 1)})
    assert com(kps) == (5.0, 5.0)

def test_torso_angle_upright_near_zero():
    # shoulders above hips (smaller y), vertical torso
    kps = _kps({5: (0, 0, 1), 6: (10, 0, 1), 11: (0, 100, 1), 12: (10, 100, 1)})
    assert torso_angle_deg(kps) < 10

def test_torso_angle_lying_near_ninety():
    # shoulders and hips at same height, horizontal torso
    kps = _kps({5: (0, 50, 1), 6: (0, 55, 1), 11: (100, 50, 1), 12: (100, 55, 1)})
    assert torso_angle_deg(kps) > 80

def test_head_below_hips_true_when_nose_lower_on_screen():
    kps = _kps({0: (5, 120, 1), 11: (0, 100, 1), 12: (10, 100, 1)})
    assert head_below_hips(kps) is True

def test_visible_count_and_core_visible():
    kps = _kps({5: (0, 0, 0.9), 6: (1, 0, 0.9), 11: (0, 1, 0.9), 12: (1, 1, 0.9)})
    assert visible_count(kps, thr=0.2) == 4
    assert core_visible(kps, thr=0.5) is True

def test_normalize_divides_by_bbox_height():
    assert normalize(30, 60) == 0.5
    assert normalize(30, 0) == 0.0