from src.improvement_recommender import ImprovementRecommender
from src.models import ConditionBalanceDiagnostic
def test_recommendation_categories()->None:
    b=ConditionBalanceDiagnostic(10,2,8,1,.5,.2,7,8.,False,"bad","add");cats={x.category for x in ImprovementRecommender().recommend(.3,.6,.7,b,8,1,True)};assert {"camera_position","face_detection","condition_balance","sample_size"}<=cats