def test_run():
run_pipeline(os.path.join(test_data_path, "test_data_json_stream.json"),
content_field="abstract",
termite_plot=True,
lda_vis=False)
assert os.path.isfile("termite.html")
os.remove("termite.html")
def run(data, format, output, tokenizer, vectorizer, ntopics, field, model,
termite, lda_vis):
run_pipeline(data_source=data,
source_type=format,
dir_path=output,
tokenizer=tokenizer,
vectorizer=vectorizer,
ntopics=ntopics,
content_field=field,
model=model,
lda_vis=lda_vis,
termite_plot=termite)
def test_run():
run_pipeline(os.path.join(test_data_path, "test_data_json_stream.json"), content_field="abstract",
termite_plot=True, lda_vis=False)
assert os.path.isfile("termite.html")
os.remove("termite.html")
def run(data, format, output, tokenizer, vectorizer, ntopics, field, model,
termite, lda_vis):
run_pipeline(data_source=data, source_type=format, dir_path=output,
tokenizer=tokenizer, vectorizer=vectorizer, ntopics=ntopics,
content_field=field, model=model, lda_vis=lda_vis,
termite_plot=termite)
