def test_predict(self, dummy_folder, dummy_text): """ Test predict on the class :return None: """ assert clusterer.DocumentClusterer(encoder.DocumentEncoder().train( dummy_folder)).train(dummy_folder).predict(dummy_text)
def test_train(self, dummy_folder): """ Test training the class :return None: """ assert clusterer.DocumentClusterer( encoder.DocumentEncoder().train(dummy_folder)).train(dummy_folder)
def test_encode(self, dummy_folder, dummy_text): """ Test text encoding on encoder :return None: """ assert encoder.DocumentEncoder().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.DocumentEncoder().train(dummy_folder)
def test_init(self): """ Test initialization of the class :return None: """ assert encoder.DocumentEncoder()
def test_init(self): """ Test initialization the class :return None: """ assert clusterer.DocumentClusterer(encoder.DocumentEncoder())