zkxw/backend/tests/test_lab_storage_code_task_set.py

import unittest

from app.utils.lab_storage_code_task_set import match_builtin_task_set_name_by_lab_storage_code


class TestMatchBuiltinTaskSetName(unittest.TestCase):
    def test_bc_before_b(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("BC20260101"),
            "干细胞培养（内置模版）",
        )

    def test_b_only(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("B20260101"),
            "造血干样本处理（内置模版）",
        )

    def test_ubmc(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("UBMC001"),
            "UBMC冻存（内置模版）",
        )

    def test_pbmc(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("pbmc001"),
            "PBMC冻存（内置模版）",
        )

    def test_empty_and_unknown(self):
        self.assertIsNone(match_builtin_task_set_name_by_lab_storage_code(""))
        self.assertIsNone(match_builtin_task_set_name_by_lab_storage_code("XY123"))

    def test_bc_case_insensitive(self):
        self.assertEqual(
            match_builtin_task_set_name_by_lab_storage_code("bc99"),
            "干细胞培养（内置模版）",
        )


if __name__ == "__main__":
    unittest.main()