Python Unittest Test If Function Returns True