added the following line to make it work
from unittest import TestCase