def test_verbalize_header_random(test_data, capsys): """Test verbalization from NLTKInterface.""" verbalization_log_file = os.path.join(LOG_FOLDER, "verbalization_random_1.logs") verbalization_log = FileWriter(verbalization_log_file) verbalization_log_time_file = os.path.join( LOG_FOLDER, "verbalization_random_time_1.logs") verbalization_log_time = FileWriter(verbalization_log_time_file) error_log_file = os.path.join(LOG_FOLDER, "verbalization_random_error.logs") error_log = FileWriter(error_log_file) start_time = time.time() empty_line = "\n" for index, header in enumerate(test_data): verbalization_log.write(repr(header)) verbalization_log.write(empty_line) try: verbalized_header = nltk.verbalize_header_random(header) verbalization_log.write(repr(verbalized_header)) verbalization_log.write(empty_line) verbalization_log.write(empty_line) except: error_log.write(repr(header)) continue end_time = time.time() verbalization_log_time.write("%s" % (end_time - start_time)) verbalization_log.close() verbalization_log_time.close()
def test_verbalize_header_palmetto(test_data, capsys): """Test verbalization from NLTKInterface.""" verbalization_log_file = os.path.join( LOG_FOLDER, "verbalization_palmetto.logs" ) verbalization_log = FileWriter(verbalization_log_file) verbalization_log_time_file = os.path.join( LOG_FOLDER, "verbalization_palmetto_time.logs" ) verbalization_log_time = FileWriter(verbalization_log_time_file) start_time = time.time() empty_line = "\n" for header in test_data: verbalization_log.write(repr(header)) verbalization_log.write(empty_line) verbalized_header = nltk.verbalize_header_palmetto(header) verbalization_log.write(repr(verbalized_header)) verbalization_log.write(empty_line) verbalization_log.write(empty_line) end_time = time.time() verbalization_log_time.write("%s" % (end_time - start_time)) verbalization_log.close() verbalization_log_time.close()
def test_get_header_synsets(header): """Test get_header_synsets method.""" synset_packs = nltk.get_header_synsets(header) for index, synset_pack in enumerate(synset_packs): (label, synsets) = synset_pack assert header[index] == label assert isinstance(synsets, list)
def test_is_graph_converge_pass(subgraph_converge): """ Test is_graph_converge method. Subgraph should converge. """ is_converge = nltk.is_graph_converge(subgraph_converge, 5) assert is_converge is True
def test_is_graph_converge_fail(subgraph_does_not_converge): """ Test is_graph_converge method. Subgraph should not converge. """ is_converge = nltk.is_graph_converge(subgraph_does_not_converge, 5) assert is_converge is False
def test_build_weighted_graph(synset_packs): """Test build_weigthed_graph method.""" synset_graph = nltk.build_weighted_graph(synset_packs) number_of_nodes = len(synset_packs) number_of_edges = number_of_nodes * (number_of_nodes - 1) / 2 assert len(synset_graph) == number_of_edges for edge in synset_graph: assert len(edge) != 0
def test_verbalize_random_fast(test_data): """Test verbalization from NLTKInterface.""" verbalization_log_file = os.path.join(LOG_FOLDER, "verbalization_fast.logs") verbalization_log = FileWriter(verbalization_log_file) start_time = time.time() empty_line = "\n" for header in test_data: verbalization_log.write(repr(header)) verbalization_log.write(empty_line) verbalized_header = nltk.verbalize_header_random(header) verbalization_log.write(repr(verbalized_header)) verbalization_log.write(empty_line) verbalization_log.write(empty_line) end_time = time.time()
def test_verbalize_random_fast(test_data): """Test verbalization from NLTKInterface.""" verbalization_log_file = os.path.join( LOG_FOLDER, "verbalization_fast.logs" ) verbalization_log = FileWriter(verbalization_log_file) start_time = time.time() empty_line = "\n" for header in test_data: verbalization_log.write(repr(header)) verbalization_log.write(empty_line) verbalized_header = nltk.verbalize_header_random(header) verbalization_log.write(repr(verbalized_header)) verbalization_log.write(empty_line) verbalization_log.write(empty_line) end_time = time.time()
def test_verbalize_header_palmetto(test_data, capsys): """Test verbalization from NLTKInterface.""" verbalization_log_file = os.path.join(LOG_FOLDER, "verbalization_palmetto.logs") verbalization_log = FileWriter(verbalization_log_file) verbalization_log_time_file = os.path.join( LOG_FOLDER, "verbalization_palmetto_time.logs") verbalization_log_time = FileWriter(verbalization_log_time_file) start_time = time.time() empty_line = "\n" for header in test_data: verbalization_log.write(repr(header)) verbalization_log.write(empty_line) verbalized_header = nltk.verbalize_header_palmetto(header) verbalization_log.write(repr(verbalized_header)) verbalization_log.write(empty_line) verbalization_log.write(empty_line) end_time = time.time() verbalization_log_time.write("%s" % (end_time - start_time)) verbalization_log.close() verbalization_log_time.close()
def test_verbalize_header_random(test_data, capsys): """Test verbalization from NLTKInterface.""" verbalization_log_file = os.path.join( LOG_FOLDER, "verbalization_random_1.logs" ) verbalization_log = FileWriter(verbalization_log_file) verbalization_log_time_file = os.path.join( LOG_FOLDER, "verbalization_random_time_1.logs" ) verbalization_log_time = FileWriter(verbalization_log_time_file) error_log_file = os.path.join( LOG_FOLDER, "verbalization_random_error.logs" ) error_log = FileWriter(error_log_file) start_time = time.time() empty_line = "\n" for index, header in enumerate(test_data): verbalization_log.write(repr(header)) verbalization_log.write(empty_line) try: verbalized_header = nltk.verbalize_header_random(header) verbalization_log.write(repr(verbalized_header)) verbalization_log.write(empty_line) verbalization_log.write(empty_line) except: error_log.write(repr(header)) continue end_time = time.time() verbalization_log_time.write("%s" % (end_time - start_time)) verbalization_log.close() verbalization_log_time.close()
def synset_graph(): """Load synset graph fixture.""" header = ['label', 'type', 'subject', 'homepage', 'foundation'] synset_packs = nltk.get_header_synsets(header) return nltk.build_weighted_graph(synset_packs)
def test_data(): """Load test data fixture.""" return nltk.load_test_data()
import opentablebench.NLTKInterface as nltk header = ['label', 'type', 'subject'] _synsets = nltk.get_header_synsets(header) import ipdb ipdb.set_trace()
import opentablebench.NLTKInterface as nltk header = ['label', 'type', 'subject'] _synsets = nltk.get_header_synsets(header) import ipdb; ipdb.set_trace()
def synset_packs(): """Load header fixture.""" header = ['label', 'type', 'subject', 'homepage', 'foundation'] return nltk.get_header_synsets(header)