def test_predict(self, dummy_folder, dummy_text):
""" Test predict on the class
:return None:
"""
assert clusterer.LineClusterer(encoder.LineEncoder().train(
dummy_folder)).train(dummy_folder).predict(dummy_text)
def test_train(self, dummy_folder):
""" Test training the class
:return None:
"""
assert clusterer.LineClusterer(
encoder.LineEncoder().train(dummy_folder)).train(dummy_folder)
def test_encode(self, dummy_folder, dummy_text):
""" Test text encoding on encoder
:return None:
"""
assert encoder.LineEncoder().train(dummy_folder).encode(
tokenizer.WhitespaceTokenizer().transform(dummy_text)).any()
def __init__(self):
self.document_encoder = encoder.DocumentEncoder()
self.line_encoder = encoder.LineEncoder()
self.document_clusterer = clusterer.DocumentClusterer(
self.document_encoder)
self.line_clusterer = clusterer.LineClusterer(self.line_encoder)
def test_train(self, dummy_folder):
""" Test training the encoder
:return None:
"""
assert encoder.LineEncoder().train(dummy_folder)
def test_init(self):
""" Test initialization of the class
:return None:
"""
assert encoder.LineEncoder()
def test_init(self):
""" Test initialization the class
:return None:
"""
assert clusterer.LineClusterer(encoder.LineEncoder())