303171d1ce88637b73a08da412f221a5bc30410c,tests/structure/test_density.py,,test_density_weights,#Any#Any#,68

Before Change

    assert density[1,1] == 2

    // weights should not be repeated along stack dimension for NxM
    stack_weights = np.array([atomic_weights, atomic_weights])
    density, (x, y, z) = struc.density(stack, weights=stack_weights)
    assert density.sum() == 11
    assert density[0,2] == 4
    assert density[1,0] == 5
    assert density[1,1] == 2

After Change

    // weights should be repeated along stack dimensions
    density, (x, y, z) = struc.density(stack, weights=atomic_weights)
    density2, (x, y, z) = struc.density(stack,
                            weights=np.array([atomic_weights, atomic_weights]))
    assert density.sum() == density2.sum() == 11
    assert density[0,2] == density2[0,2] == 4
    assert density[1,0] == density2[1,0] == 5
    assert density[1,1] == density2[1,1] == 2

Instances