Complete test_weight_decay_0 test

Signed-off-by: Axel Davy <axel.davy@ens.fr>