5 lines
396 B
Python
5 lines
396 B
Python
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
|