def loadTestSuite(input_file, bbox=False, k=5):
TS = defaultdict()
with open(input_file) as fin:
tcID = 1
for tc in fin:
if bbox:
TS[tcID] = tc[:-1]
else:
TS[tcID] = set(tc[:-1].split())
tcID += 1
shuffled = list(TS.keys())
random.shuffle(shuffled)
newTS = OrderedDict()
for key in shuffled:
newTS[key] = TS[key]
if bbox:
newTS = lsh.kShingles(TS, k)
return newTS
def loadTestSuite(input_file, bbox=False, k=5):
"""INPUT
(str)input_file: path of input file
OUTPUT
(dict)TS: key=tc_ID, val=set(covered lines)
"""
TS = defaultdict()
with open(input_file) as fin:
tcID = 1
for tc in fin:
if bbox:
TS[tcID] = tc[:-1]
else:
TS[tcID] = set(tc[:-1].split())
tcID += 1
shuffled = TS.items()
random.shuffle(shuffled)
TS = OrderedDict(shuffled)
if bbox:
TS = lsh.kShingles(TS, k)
return TS
