Switch unittest for unittest2