mirror of
https://github.com/datahub-project/datahub.git
synced 2025-08-06 16:28:16 +00:00
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
|
from datahub.ingestion.glossary.classification_mixin import (
|
||
|
|
ClassificationHandler,
|
||
|
|
ClassificationReportMixin,
|
||
|
|
ClassificationSourceConfigMixin,
|
||
|
|
)
|
||
|
|
from datahub.ingestion.glossary.classifier import ClassificationConfig
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_classifiers_without_classification_config():
|
||
|
|
# Create a config without classification attribute
|
||
|
|
class TestConfig(ClassificationSourceConfigMixin):
|
||
|
|
classification: ClassificationConfig = None # type: ignore
|
||
|
|
|
||
|
|
config = TestConfig()
|
||
|
|
report = ClassificationReportMixin()
|
||
|
|
|
||
|
|
# Create handler with config that has no classification attribute
|
||
|
|
handler = ClassificationHandler(config, report)
|
||
|
|
|
||
|
|
# Should return empty list when classification is not set
|
||
|
|
assert handler.get_classifiers() == []
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_classifiers_with_none_classification():
|
||
|
|
# Create a config with classification set to None
|
||
|
|
class TestConfig(ClassificationSourceConfigMixin):
|
||
|
|
classification: ClassificationConfig = None # type: ignore
|
||
|
|
|
||
|
|
config = TestConfig()
|
||
|
|
report = ClassificationReportMixin()
|
||
|
|
|
||
|
|
# Create handler with config that has None classification
|
||
|
|
handler = ClassificationHandler(config, report)
|
||
|
|
|
||
|
|
# Should return empty list when classification is None
|
||
|
|
assert handler.get_classifiers() == []
|