import unittest
from typing import List

from pymilvus import FieldSchema, DataType

from app.models.schp.simple_extractor import get_color_by_label
from app.services.milvus import MilvusClient


class TestGetColorByLabel(unittest.TestCase):
    def test_get_color_by_label(self):
        print(get_color_by_label("Jumpsuits"))

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