Improve tests

[codecov[bot]]

Codecov Report

Patch coverage: 100.00% and project coverage change: +0.31% :tada:

Comparison is base (d247249) 98.75% compared to head (7ca1d01) 99.06%.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #8 +/- ## ========================================== + Coverage 98.75% 99.06% +0.31% ========================================== Files 10 10 Lines 963 962 -1 ========================================== + Hits 951 953 +2 + Misses 12 9 -3 ``` | [Files Changed](https://app.codecov.io/gh/juliendenize/torchaug/pull/8?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Julien+Denize) | Coverage Δ | | |---|---|---| | [torchaug/batch\_transforms/transforms.py](https://app.codecov.io/gh/juliendenize/torchaug/pull/8?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Julien+Denize#diff-dG9yY2hhdWcvYmF0Y2hfdHJhbnNmb3Jtcy90cmFuc2Zvcm1zLnB5) | `99.68% <100.00%> (+0.62%)` | :arrow_up: | | [torchaug/transforms/functional.py](https://app.codecov.io/gh/juliendenize/torchaug/pull/8?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Julien+Denize#diff-dG9yY2hhdWcvdHJhbnNmb3Jtcy9mdW5jdGlvbmFsLnB5) | `100.00% <100.00%> (ø)` | | | [torchaug/transforms/transforms.py](https://app.codecov.io/gh/juliendenize/torchaug/pull/8?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Julien+Denize#diff-dG9yY2hhdWcvdHJhbnNmb3Jtcy90cmFuc2Zvcm1zLnB5) | `99.53% <100.00%> (ø)` | | ... and [1 file with indirect coverage changes](https://app.codecov.io/gh/juliendenize/torchaug/pull/8/indirect-changes?src=pr&el=tree-more&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Julien+Denize)

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.