def test_shortest(self):
parse = DependencyParse(self.dep, enhanced=True)
shortest = parse.get_path_from_parse(2, 4, 5, 6)
self.assertEqual(shortest, [
[self.words.index('curry'), self.words.index('had'), 'dobj_from'],
[self.words.index('had'), None, 'root'],
[self.words.index('had'), self.words.index('Curry'), 'nsubj_to'],
])
def test_shorest_real(self):
import csv
from data.adaptors import KBPDataAdaptor
with open(os.path.join(mydir, '..', 'data', 'raw', 'test.sample.tsv')) as f:
reader = csv.reader(f, delimiter="\t")
raw = reader.next()
adaptor = KBPDataAdaptor()
ex = adaptor.to_example(raw)
parse = DependencyParse(ex.dependency, enhanced=True)
shortest = parse.get_path_from_parse(ex.subject_begin, ex.subject_end, ex.object_begin, ex.object_end)
self.assertEqual(shortest, [
[ex.words.index('berkeley'), ex.words.index('baker'), 'nmod:in_from'],
])
