Ejemplo n.º 1
0
def test_format():
    features = make_features_with_class_confidence_columns()
    encoding = FormatStringEncoding(format='{class}: {confidence:.2f}')

    values = encoding(features)

    np.testing.assert_array_equal(values, ['a: 0.50', 'b: 1.00', 'c: 0.25'])
Ejemplo n.º 2
0
def test_validate_from_format_string():
    argument = '{class}: {score:.2f}'
    expected = FormatStringEncoding(format=argument)

    actual = validate_string_encoding(argument)

    assert actual == expected
Ejemplo n.º 3
0
def test_validate_from_format_dict():
    format = '{class}: {score:.2f}'
    argument = {'format': format}
    expected = FormatStringEncoding(format=format)

    actual = validate_string_encoding(argument)

    assert actual == expected
Ejemplo n.º 4
0
def test_format_with_missing_field():
    features = make_features_with_class_confidence_columns()
    encoding = FormatStringEncoding(format='{class}: {score:.2f}')

    with pytest.raises(KeyError):
        encoding(features)
Ejemplo n.º 5
0
def test_format_with_bad_string():
    features = make_features_with_class_confidence_columns()
    encoding = FormatStringEncoding(format='{class}: {confidence:.2f')

    with pytest.raises(ValueError):
        encoding(features)
Ejemplo n.º 6
0
def test_format(features):
    encoding = FormatStringEncoding(format='{class}: {confidence:.2f}')
    values = encoding(features)
    np.testing.assert_array_equal(values, ['a: 0.50', 'b: 1.00', 'c: 0.25'])
Ejemplo n.º 7
0
def test_format_with_missing_field(features):
    encoding = FormatStringEncoding(format='{class}: {score:.2f}')
    with pytest.raises(KeyError):
        encoding(features)
Ejemplo n.º 8
0
def test_format_with_bad_string(features):
    encoding = FormatStringEncoding(format='{class}: {confidence:.2f')
    with pytest.raises(ValueError):
        encoding(features)