pep8 and assert_equal instead of assert