Use assertTrue instead of assertGreater in unittest.