예제 #1
0
def test_descriptions():
    primitives = {
        NumCharacters: "Calculates the number of characters in a string.",
        Day: "Determines the day of the month from a datetime.",
        Last: "Determines the last value in a list.",
        GreaterThan: "Determines if values in one list are greater than another list.",
    }
    assert _get_descriptions(list(primitives.keys())) == list(primitives.values())
예제 #2
0
def test_descriptions():
    primitives = {
        NumCharacters: 'Return the characters in a given string.',
        Day: 'Transform a Datetime feature into the day.',
        Last: 'Returns the last value.',
        GreaterThan: ''
    }
    assert _get_descriptions(list(primitives.keys())) == list(
        primitives.values())
def test_descriptions():
    primitives = {
        NumCharacters: 'Calculates the number of characters in a string.',
        Day: 'Determines the day of the month from a datetime.',
        Last: 'Returns the last value.',
        GreaterThan: ''
    }
    assert _get_descriptions(list(primitives.keys())) == list(
        primitives.values())
예제 #4
0
def test_list_primitives_order():
    df = list_primitives()
    all_primitives = get_transform_primitives()
    all_primitives.update(get_aggregation_primitives())

    for name, primitive in all_primitives.items():
        assert name in df['name'].values
        row = df.loc[df['name'] == name].iloc[0]
        actual_desc = _get_descriptions([primitive])[0]
        if actual_desc:
            assert actual_desc == row['description']

    types = df['type'].values
    assert 'aggregation' in types
    assert 'transform' in types
예제 #5
0
def test_list_primitives_order():
    df = list_primitives()
    all_primitives = get_transform_primitives()
    all_primitives.update(get_aggregation_primitives())

    for name, primitive in all_primitives.items():
        assert name in df["name"].values
        row = df.loc[df["name"] == name].iloc[0]
        actual_desc = _get_descriptions([primitive])[0]
        if actual_desc:
            assert actual_desc == row["description"]
        assert row["dask_compatible"] == (Library.DASK in primitive.compatibility)
        assert row["valid_inputs"] == ", ".join(
            _get_unique_input_types(primitive.input_types)
        )
        assert row["return_type"] == getattr(primitive.return_type, "__name__", None)

    types = df["type"].values
    assert "aggregation" in types
    assert "transform" in types
def test_list_primitives_order():
    df = list_primitives()
    all_primitives = get_transform_primitives()
    all_primitives.update(get_aggregation_primitives())

    for name, primitive in all_primitives.items():
        assert name in df['name'].values
        row = df.loc[df['name'] == name].iloc[0]
        actual_desc = _get_descriptions([primitive])[0]
        if actual_desc:
            assert actual_desc == row['description']
        assert row['dask_compatible'] == (Library.DASK
                                          in primitive.compatibility)
        assert row['valid_inputs'] == ', '.join(
            _get_names_valid_inputs(primitive.input_types))
        assert row['return_type'] == getattr(primitive.return_type, '__name__',
                                             None)

    types = df['type'].values
    assert 'aggregation' in types
    assert 'transform' in types