Updated 2.6 test to use unittest2