Use py26 compatible import for unittest

So that we get unittest2 assertion methods
(specifically assertIn for the tests that are failing in py26).