53 lines
2.2 KiB
Python
53 lines
2.2 KiB
Python
"""Классификация кормов по группам для авторациона."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from app import create_app, db
|
|
from app.lab.calc.feed_groups import classify_feed_group, validate_group_selections
|
|
from app.models import Component
|
|
from app.models.base import default_uuid
|
|
from app.services.setup_state import mark_setup_complete
|
|
from tests.helpers.zootech_test_helpers import ZootechTestConfig
|
|
|
|
|
|
class LabFeedGroupsTests(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
cls.app = create_app(ZootechTestConfig)
|
|
cls.ctx = cls.app.app_context()
|
|
cls.ctx.push()
|
|
db.create_all()
|
|
mark_setup_complete(cls.app)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls) -> None:
|
|
db.session.remove()
|
|
db.drop_all()
|
|
cls.ctx.pop()
|
|
|
|
def _comp(self, name: str, ctype: str) -> Component:
|
|
c = Component(id=default_uuid(), name=name, type=ctype, dry_matter=80.0, price=10.0)
|
|
db.session.add(c)
|
|
db.session.commit()
|
|
return c
|
|
|
|
def test_classify_by_canonical_type(self) -> None:
|
|
self.assertEqual(classify_feed_group(self._comp("Сено", "Грубые корма")), "rough")
|
|
self.assertEqual(classify_feed_group(self._comp("Силос", "Сочные корма")), "succulent")
|
|
self.assertEqual(classify_feed_group(self._comp("Комбикорм", "Концентрированные")), "concentrate")
|
|
self.assertEqual(classify_feed_group(self._comp("Премикс", "Добавки")), "other")
|
|
|
|
def test_classify_legacy_type_fallback(self) -> None:
|
|
self.assertEqual(classify_feed_group(self._comp("Комбикорм", "зерновые")), "concentrate")
|
|
self.assertEqual(classify_feed_group(self._comp("Сено", "объемные корма")), "rough")
|
|
|
|
def test_validate_requires_rough_and_three_total(self) -> None:
|
|
errs = validate_group_selections({"rough": ["a"], "succulent": [], "concentrate": [], "other": []})
|
|
self.assertIn("pool:need_3", errs)
|
|
errs2 = validate_group_selections(
|
|
{"rough": ["a"], "succulent": ["b"], "concentrate": ["c"], "other": []}
|
|
)
|
|
self.assertEqual(errs2, [])
|