MemOS/tests/configs/test_parser.py at main · rontian/MemOS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from memos.configs.parser import BaseParserConfig, MarkItDownParserConfig, ParserConfigFactory
from tests.utils import (
    check_config_base_class,
    check_config_factory_class,
    check_config_instantiation_invalid,
    check_config_instantiation_valid,
)


def test_base_parser_config():
    check_config_base_class(
        BaseParserConfig,
        required_fields=[],
        optional_fields=[],
    )

    check_config_instantiation_valid(
        BaseParserConfig,
        {},
    )

    check_config_instantiation_invalid(BaseParserConfig)


def test_markitdown_parser_config():
    check_config_base_class(
        MarkItDownParserConfig,
        required_fields=[],
        optional_fields=[],
    )

    check_config_instantiation_valid(
        MarkItDownParserConfig,
        {},
    )

    check_config_instantiation_invalid(MarkItDownParserConfig)


def test_parser_config_factory():
    check_config_factory_class(
        ParserConfigFactory,
        expected_backends=["markitdown"],
    )

    check_config_instantiation_valid(
        ParserConfigFactory,
        {
            "backend": "markitdown",
            "config": {},
        },
    )

    check_config_instantiation_invalid(ParserConfigFactory)