add pruning tests on 4 models

[cregouby]

Now that https://github.com/mlverse/torch/pull/819 is on cran, it's time to integrate the pruning test in the relevant packages.