python: add missing header

Signed-off-by: Martin Milata <mmilata@redhat.com>
tests: do not use assertIn in python tests

The method was added in python 2.7 which is not available on rhel6.

Signed-off-by: Martin Milata <mmilata@redhat.com>