OCR test sample

import os
import sys
import base64
import requests
import json

# config host address
HOST = "http://localhost:10000"

def main():
    num_pass = 0
    num_fail = 0

    with open("./groundtruth.json",'r') as gt_file:
        files = json.load(gt_file)

        for f in files:
            print(f["filename"])
            byimage = b"data:application/octet-stream;base64," + base64.b64encode(open('./imgs/'+f["filename"]).read())
            headers = {"Content-Type": "application/json", "Authorization": "QiniuStub uid=1&ut=2"}
            r = requests.post(HOST + "/v1/ocr/text", headers=headers, data=json.dumps({"data":{"uri":byimage.decode('utf8')}}))
            print r.text
            r = json.loads(r.text)
            if json.dumps(r["result"]) == json.dumps(f["result"]):
                print('test file ' + f["filename"] + ': pass')
                num_pass += 1
            else:
                print('test file ' + f["filename"] + ': failed')
                num_fail += 1

    print()
    print(str(num_pass) + ' pass')
    print(str(num_fail) + ' failed')

if __name__ == '__main__':
    main()