import os, sys
from SIFT import SIFT
from retrieval import retrieval
from pprint import pprint


def main():
    print("start")
    # img = cv.imread("beach.jpg")
    # cv.imshow("Image", img)
    # from matplotlib import pyplot as plt
    # plt.hist(img.ravel(),10,[0,256]); plt.show()
    # gray_img = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
    # cv.imshow("Gray Image", gray_img)
    # cv.waitKey()

    # for para_test in [8, 16]:
    #     print("test using " + str(para_test))
    #     assert retrieval("2") == "2"
    #     print("test using " + str(para_test) + " passed")

    # flower
    # assert retrieval("6") == "7"  # 731, horse
    # assert retrieval("1") == "1"  # 126, 138, beach
    # assert retrieval("4") == "4"  # 458, dinosaur
    # assert retrieval("5") == "6"  # 6xx means flower category
    #
    assert retrieval("2") == "2"  # 2xx means building
    #
    # assert retrieval('3') == "3"

    print("test done")
    sys.exit()


main()