Update test.py to use unittest
This allows for:
* test filtering using -k
* parallel testing using pytest
* verbose output of what is being run
* other timing functionalities that pytest provides
ghstack-source-id: c3d2f64f951a22d802b5cd8f2a00e758682b8b21
Pull Request resolved: https://github.com/pytorch/benchmark/pull/81
