Try to import unittest2 instead of unittest if possible