[Tracker] Split tests into multiple file tests

[johnnv1]

Currently, we have some test files that have all (or almost all) tests of a module within a single file, the ideal here is to split these cases into multiple test files. Which file should be testing a single functionality, or file of our library.

Example:

We have now a single file testing almost all losses functions https://github.com/kornia/kornia/blob/c5558ac2c03bbbbbbb6c9f82616b8d5229398f1a/tests/losses/test_losses.py we would like to have a single test file for each loss. What would mirror the file organization of the losses module of kornia https://github.com/kornia/kornia/tree/c5558ac2c03bbbbbbb6c9f82616b8d5229398f1a/kornia/losses

This issue will work as a tracker to map some of these files that need to be split, feel free to help us find others and/or help us split them into multiple files.

The goal here is to reorganize our test suite. If you have any questions, you can call us and/or ask on our Slack page too

Originally posted by @edgarriba in #2745 (comment)

---

You can choose anyone below, or find new ones. You don't have to worry about separating all the tests in a file into a single PR

• Split losses tests - https://github.com/kornia/kornia/blob/ce434e467faf617604bb3383cf78cd0b79f59dbd/tests/losses/test_losses.py - Test suite: separate losses tests into individual files #2801
• Split metrics tests - https://github.com/kornia/kornia/blob/ce434e467faf617604bb3383cf78cd0b79f59dbd/tests/test_metrics.py
• Split contrib tests - https://github.com/kornia/kornia/blob/ce434e467faf617604bb3383cf78cd0b79f59dbd/tests/contrib/test_contrib.py - Test suite: separate contrib tests into individual files #2802
• Split augmentations tests
  • 2d - https://github.com/kornia/kornia/blob/ce434e467faf617604bb3383cf78cd0b79f59dbd/tests/augmentation/test_augmentation.py
  • 3d - https://github.com/kornia/kornia/blob/ce434e467faf617604bb3383cf78cd0b79f59dbd/tests/augmentation/test_augmentation_3d.py
• Split container augmentation tests - https://github.com/kornia/kornia/blob/ce434e467faf617604bb3383cf78cd0b79f59dbd/tests/augmentation/test_container.py

[marvlyngkhoi]

@johnnv1 I'll help splitting the losses test into their individual loss file and can you help point me to the testing docs so that I can run the tests after I've done splitting the file thanks.

[marvlyngkhoi]

@johnnv1 I have split the test losses into different inidividual files and linking the PR here for you to check once your free
https://github.com/kornia/kornia/pull/2801.