-
Notifications
You must be signed in to change notification settings - Fork 739
Closed
Description
unittest is a great starting point for checking the implementation, but I would prefer to allow pytest for testing as well. It allows an easier way to write heavily parametrized tests. See also the discussion at pytorch/pytorch#11578.
It will perfectly fine work together with unittest, so there is no need to rewrite any tests, but it would allow to add new tests using pytest. What we would have to do is to install it, e.g. add a test/requirement.txt file or use pip, basically replace
$ python -m unittestwith
$ pip install pytest # or: pip install -r test/requirements.txt
$ pytestBTW, it is used by pytorch/vision on travis, see https://github.com/pytorch/vision/blob/master/.travis.yml#L53
If you would be fine with the proposal, I could create a pull request integrating it.
Metadata
Metadata
Assignees
Labels
No labels