def test_normal():
annotator = FreebaseAnnotator(FREEBASE_DUMP_PATH, FREEBASE_RAW_PATH)
sent = 'Viacom Inc. and Apple Computers released a new phone .'.split()
actual = annotator.annotate(sent)
expected = [((0,1), VIACOM), ((0,1), VIACOM2), ((3,4), APPLE)]
assert_equal(actual, expected)
def test_empty_output():
annotator = FreebaseAnnotator(FREEBASE_DUMP_PATH, FREEBASE_RAW_PATH)
sent = 'Halo Inc. released a new phone .'.split()
actual = annotator.annotate(sent)
expected = []
assert_equal(actual, expected)
