Use pytest instead of nose

This MR moves unit testing to pytest (instead of nose). It incidentally also enables py36 builds, now that pytorch is available for that Python version.