ludwig-ai--ludwig
wehub-resource-sync
593b94c120
pytest / Unit Tests (push) Has been cancelled
pytest / Integration (integration_tests_a) (push) Has been cancelled
pytest / Integration (integration_tests_b) (push) Has been cancelled
pytest / Integration (integration_tests_c) (push) Has been cancelled
pytest / Integration (integration_tests_d) (push) Has been cancelled
pytest / Integration (integration_tests_e) (push) Has been cancelled
pytest / Integration (integration_tests_f) (push) Has been cancelled
pytest / Integration (integration_tests_g) (push) Has been cancelled
pytest / Integration (integration_tests_h) (push) Has been cancelled
pytest / Integration (integration_tests_i) (push) Has been cancelled
pytest / Integration (integration_tests_j) (push) Has been cancelled
pytest / Distributed (distributed_a) (push) Has been cancelled
pytest / Distributed (distributed_b) (push) Has been cancelled
pytest / Distributed (distributed_c) (push) Has been cancelled
pytest / Distributed (distributed_d) (push) Has been cancelled
pytest / Distributed (distributed_e) (push) Has been cancelled
pytest / Distributed (distributed_f) (push) Has been cancelled
pytest / Minimal Install (push) Has been cancelled
pytest / Event File (push) Has been cancelled
pytest (slow) / py-slow (push) Has been cancelled
Publish JSON Schema / publish-schema (push) Has been cancelled
24 行
1.1 KiB
Python
24 行
1.1 KiB
Python
#! /usr/bin/env python
|
|
|
|
|
|
from ludwig.schema.features.preprocessing.binary import BinaryPreprocessingConfig
|
|
from ludwig.schema.features.preprocessing.category import CategoryPreprocessingConfig
|
|
from ludwig.schema.features.preprocessing.utils import PreprocessingDataclassField
|
|
|
|
|
|
def get_marshmallow_from_dataclass_field(dfield):
|
|
"""Helper method for checking marshmallow metadata succinctly."""
|
|
return dfield.metadata["marshmallow_field"]
|
|
|
|
|
|
def test_preprocessing_dataclass_field():
|
|
binary_preproc_dataclass = PreprocessingDataclassField("binary")
|
|
assert binary_preproc_dataclass.default_factory is not None
|
|
assert get_marshmallow_from_dataclass_field(binary_preproc_dataclass).allow_none is False
|
|
assert binary_preproc_dataclass.default_factory() == BinaryPreprocessingConfig()
|
|
|
|
category_preproc_dataclass = PreprocessingDataclassField("category")
|
|
assert category_preproc_dataclass.default_factory is not None
|
|
assert get_marshmallow_from_dataclass_field(category_preproc_dataclass).allow_none is False
|
|
assert category_preproc_dataclass.default_factory() == CategoryPreprocessingConfig()
|