FastMoE test === To run unit test, directly run `pytest` in this directory. `test.sh` is a wrapper script to execute single tests without pytest for debugging purpose.