Add support for testing with pymysql