def test_machine_comprehension(self):
predictor = pretrained.bidirectional_attention_flow_seo_2017()
passage = """The Matrix is a 1999 science fiction action film written and directed by The Wachowskis, starring Keanu Reeves, Laurence Fishburne, Carrie-Anne Moss, Hugo Weaving, and Joe Pantoliano. It depicts a dystopian future in which reality as perceived by most humans is actually a simulated reality called "the Matrix", created by sentient machines to subdue the human population, while their bodies' heat and electrical activity are used as an energy source. Computer programmer Neo" learns this truth and is drawn into a rebellion against the machines, which involves other people who have been freed from the "dream world". """ # pylint: disable=line-too-long
question = "Who stars in The Matrix?"
result = predictor.predict_json({"passage": passage, "question": question})
correct = "Keanu Reeves, Laurence Fishburne, Carrie-Anne Moss, Hugo Weaving, and Joe Pantoliano"
assert correct == result["best_span_str"]
def __init__(self):
"""
During initialization, Allen's BiDAF model is setup
"""
self.model = pretrained.bidirectional_attention_flow_seo_2017()