Function assert_sparse

tests/conftest.py:423–435

Test that sparse output is created upon request in the correct format.

(descriptor_func)

def assert_sparse(descriptor_func):
    """Test that sparse output is created upon request in the correct format."""
    system = get_simple_finite()

    # Dense
    desc = descriptor_func(sparse=False)([system])
    features = desc.create(system)
    assert type(features) == np.ndarray

    # Sparse
    desc = descriptor_func(sparse=True)([system])
    features = desc.create(system)
    assert type(features) == sparse.COO


def assert_n_features(descriptor_func, n_features):