Code Duplication    Length = 12-12 lines in 2 locations

tests/milvus_benchmark/runner.py 2 locations

@@ 73-84 (lines=12) @@
70
    # use the first file, nq should be less than VECTORS_PER_FILE
71
    if nq > MAX_NQ:
72
        raise Exception("Over size nq")
73
    if data_type == "random":
74
        file_name = SRC_BINARY_DATA_DIR+'query_%d.npy' % dimension
75
    elif data_type == "sift":
76
        file_name = SIFT_SRC_DATA_DIR+'query.npy'
77
    elif data_type == "deep":
78
        file_name = DEEP_SRC_DATA_DIR+'query.npy'
79
    elif data_type == "jaccard":
80
        file_name = JACCARD_SRC_DATA_DIR+'query.npy'
81
    elif data_type == "hamming":
82
        file_name = HAMMING_SRC_DATA_DIR+'query.npy'
83
    elif data_type == "sub" or data_type == "super":
84
        file_name = STRUCTURE_SRC_DATA_DIR+'query.npy'
85
    data = np.load(file_name)
86
    vectors = data[0:nq].tolist()
87
    return vectors
@@ 54-65 (lines=12) @@
51
def gen_file_name(idx, dimension, data_type):
52
    s = "%05d" % idx
53
    fname = FILE_PREFIX + str(dimension) + "d_" + s + ".npy"
54
    if data_type == "random":
55
        fname = SRC_BINARY_DATA_DIR+fname
56
    elif data_type == "sift":
57
        fname = SIFT_SRC_DATA_DIR+fname
58
    elif data_type == "deep":
59
        fname = DEEP_SRC_DATA_DIR+fname
60
    elif data_type == "jaccard":
61
        fname = JACCARD_SRC_DATA_DIR+fname
62
    elif data_type == "hamming":
63
        fname = HAMMING_SRC_DATA_DIR+fname
64
    elif data_type == "sub" or data_type == "super":
65
        fname = STRUCTURE_SRC_DATA_DIR+fname
66
    return fname
67
68