64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
from datetime import date
|
|
|
|
import pytest
|
|
|
|
from app.calculator.core import calculate_saju
|
|
from app.calculator.analysis import perform_full_analysis
|
|
from app.calculator.lucky import calculate_lucky
|
|
|
|
|
|
def _saju_for(year, month, day, hour, gender):
|
|
saju = calculate_saju(year, month, day, hour, gender)
|
|
analysis = perform_full_analysis(saju, 2026)
|
|
return saju, analysis
|
|
|
|
|
|
def test_lucky_keys():
|
|
saju, analysis = _saju_for(1990, 5, 15, 14, "male")
|
|
r = calculate_lucky(saju, analysis, date(2026, 5, 26))
|
|
for k in ("color", "number", "direction", "good_signs", "warnings"):
|
|
assert k in r, f"missing {k}"
|
|
|
|
|
|
def test_lucky_number_range():
|
|
saju, analysis = _saju_for(1990, 5, 15, 14, "male")
|
|
r = calculate_lucky(saju, analysis, date(2026, 5, 26))
|
|
assert 1 <= r["number"] <= 9
|
|
|
|
|
|
def test_lucky_color_from_yongshin():
|
|
saju, analysis = _saju_for(1990, 5, 15, 14, "male")
|
|
r = calculate_lucky(saju, analysis, date(2026, 5, 26))
|
|
assert isinstance(r["color"], list)
|
|
assert len(r["color"]) >= 1
|
|
yongshin = analysis["yong_shin"]["yong_shin"]
|
|
valid_colors_by_element = {
|
|
"木": {"청록", "녹색"},
|
|
"火": {"빨강", "주황"},
|
|
"土": {"황색", "베이지"},
|
|
"金": {"흰색", "은색"},
|
|
"水": {"파랑", "검정"},
|
|
}
|
|
expected_set = valid_colors_by_element[yongshin]
|
|
assert all(c in expected_set for c in r["color"])
|
|
|
|
|
|
def test_lucky_direction_from_yongshin():
|
|
saju, analysis = _saju_for(1990, 5, 15, 14, "male")
|
|
r = calculate_lucky(saju, analysis, date(2026, 5, 26))
|
|
valid_dirs = {"동쪽", "남쪽", "중앙", "서쪽", "북쪽"}
|
|
assert r["direction"] in valid_dirs
|
|
|
|
|
|
def test_good_signs_and_warnings_are_lists():
|
|
saju, analysis = _saju_for(1990, 5, 15, 14, "male")
|
|
r = calculate_lucky(saju, analysis, date(2026, 5, 26))
|
|
assert isinstance(r["good_signs"], list)
|
|
assert isinstance(r["warnings"], list)
|
|
|
|
|
|
def test_handles_missing_hour():
|
|
saju, analysis = _saju_for(1990, 5, 15, None, "male")
|
|
r = calculate_lucky(saju, analysis, date(2026, 5, 26))
|
|
assert 1 <= r["number"] <= 9
|