def TestExtNegativeSampling(): dataFile = "./dataset/samples/qa-dump-1460090355004_new.json" dataProvider = DataProcessor(dataFile) nNegSample = 100 dataProvider.NegSampleExt(nNegSample) for title in dataProvider.data.keys(): article = dataProvider.data[title] for i in range(len(article["answers"])): for negSample in article["negExtSamples"][i]: print " ".join(SentenceToWord( (article["answers"][i], ))[0]), negSample assert " ".join(SentenceToWord( (article["answers"][i], ))[0]) in negSample print "Extension negative sampling test passed!"