def test_transformation_dictionary(input_fields, derived_fields, output_fields,
expected_data_dictionary,
expected_transformation_dictionary):
converter = EstimatorConverter(DecisionTreeClassifier(),
context=TransformationContext(
input=input_fields,
derived=derived_fields,
model=input_fields + derived_fields,
output=output_fields),
mode=EstimatorConverter.MODE_CLASSIFICATION)
assert converter.data_dictionary().toxml(
) == expected_data_dictionary, 'Error in data dictionary generation'
assert converter.transformation_dictionary().toxml() == expected_transformation_dictionary,\
'Error in transformation dictionary generation'
def test_transformation_dictionary(input_fields, derived_fields, output_fields, expected_data_dictionary, expected_transformation_dictionary):
converter = EstimatorConverter(
DecisionTreeClassifier(),
context=TransformationContext({
Schema.INPUT: input_fields,
Schema.DERIVED: derived_fields,
Schema.MODEL: input_fields + derived_fields,
Schema.OUTPUT: output_fields
}),
mode=ModelMode.CLASSIFICATION
)
assert converter.data_dictionary().toxml() == expected_data_dictionary, 'Error in data dictionary generation'
assert converter.transformation_dictionary().toxml() == expected_transformation_dictionary,\
'Error in transformation dictionary generation'
def test_transformation_dictionary(
input_fields, derived_fields, output_fields, expected_data_dictionary, expected_transformation_dictionary
):
converter = EstimatorConverter(
DecisionTreeClassifier(),
context=TransformationContext(
input=input_fields, derived=derived_fields, model=input_fields + derived_fields, output=output_fields
),
mode=EstimatorConverter.MODE_CLASSIFICATION,
)
assert converter.data_dictionary().toxml() == expected_data_dictionary, "Error in data dictionary generation"
assert (
converter.transformation_dictionary().toxml() == expected_transformation_dictionary
), "Error in transformation dictionary generation"