def test_disambiguate_subject_column(): """ """ mlmodel = MLModel() table = mlmodel.get_tables()[0] entities = disambiguate_subject_column(table) import ipdb ipdb.set_trace()
from taipan.entitysearch.agdistis import AgdistisWrapper from taipan.ml.model import MLModel from taipan.sparql.propertysearch import * AGDISTIS = AgdistisWrapper() MLMODEL = MLModel() PROPERTIES_0 = { 0: { 0: [], 1: [], 2: [] }, 1: { 0: [], 1: [], 2: [] }, 2: { 0: [], 1: [], 2: [] } } PROPERTIES_1 = { 0: { 0: [], 1: [ 'http://dbpedia.org/property/placeOfDeath', 'http://dbpedia.org/ontology/populationPlace',
"""Test for taipan.agdistis""" from os.path import isfile, join from taipan.entitysearch.agdistis import AgdistisWrapper from taipan.ml.model import MLModel from taipan.generictable import GenericTable ENTITY = [{'start': 0, 'namedEntity': 'London', 'disambiguatedURL': 'http://dbpedia.org/resource/London', 'offset': 6}] ROW_ENTITIES = [['http://dbpedia.org/resource/Jeff_Jahn'], [], []] TO_FLATTEN = [{'start': 0, 'namedEntity': 'EUR', 'disambiguatedURL': 'http://dbpedia.org/resource/EUR,_Rome', 'offset': 3}] FLATTENED = ['http://dbpedia.org/resource/EUR,_Rome'] TO_FLATTEN_NO_URL = [{'start': 0, 'namedEntity': '35.00', 'disambiguatedURL': None, 'offset': 5}] agdistis_wrapper = AgdistisWrapper() mlmodel = MLModel() table = mlmodel.get_tables()[0] def test_flatten_urls(): flattened = agdistis_wrapper.flatten_urls(TO_FLATTEN) assert flattened == FLATTENED flattened = agdistis_wrapper.flatten_urls(TO_FLATTEN_NO_URL) assert flattened == [] def test_disambiguate_entity(): entity = agdistis_wrapper.disambiguate_entity("London") assert entity == ENTITY def test_disambiguate_table(): entities = agdistis_wrapper.disambiguate_table(table)
def test_disambiguate_table_subject_column_only(): mlmodel = MLModel() table = mlmodel.get_tables()[0] entities = disambiguate_table_subject_column_only(table) assert entities == TABLE_ENTITIES_SC_ONLY
def test_disambiguate_table(): mlmodel = MLModel() table = mlmodel.get_tables()[0] entities = disambiguate_table(table) assert entities == TABLE_ENTITIES
def test_get_support(): mlmodel = MLModel() table = mlmodel.get_tables()[0] support = S_CALC.get_support(table) assert support == SUPPORT