This change is incorrect, because ``TestCase`` subclasses don't work with methods with test generators (``yield`` statements). 

In this case the ``yield`` statements don't make much sense though, so removing them fixes the issue again. Please check though that the same number of tests are run with and without subclassing ``TestCase``.